프로그래머스 / 스택 , 큐 / 다리를 지나는 트럭

https://programmers.co.kr/learn/courses/30/lessons/42583

코딩테스트 연습 - 다리를 지나는 트럭

 

#include <string>
#include <vector>
#include <queue>
 
using namespace std;
 
//트럭 구조체
struct Truck
{
    //이동거리, 무게
    int moveDistance;
    int weight;
};
 
int solution(int bridge_length, int weight, vector<int> truck_weights) {
    //지난 시간 
    int time = 1;
    //현재 다리의 무게
    int bridge_weight = 0;
    //트럭의 인덱스
    int idx = 0;
    //트럭 구조체
    Truck sTruck;
    //다리를 지나고 있는 트럭들
    deque<Truck> crossing;
    //다리를 지나간 트럭들
    deque<Truck> passed;
 
    while (1)
    {
        //현재 다리의 무게 + 올라갈 트럭의 무게 <= 다리의 무게 한계치 일 경우 트럭이 올라간다.
        if (bridge_weight + truck_weights[idx] <= weight && truck_weights[idx] != NULL)
        {
            //현재 다리의 무게에 트럭의 무게를 더해준다.
            bridge_weight += truck_weights[idx];
            //구조체에 트럭의 무게를 넣고 
            sTruck.weight = truck_weights[idx];
            //이동거리를 초기화 한 후
            sTruck.moveDistance = 0;
            //crossing 디큐(다리를 지나고 있는 트럭들)에 푸쉬 백 한다.
            crossing.push_back(sTruck);
            //트럭의 인덱스를 ++시켜 다음 트럭이 들어올 수 있게 한다.
            idx++;
        }
        //다리를 지난 트럭들이 있을 경우 (예외처리)
        if (!crossing.empty())
        {
            for (int i = 0; i < crossing.size(); i++)
            {
                //다리를 지나고 있는 트럭의 이동 거리를 ++
                crossing[i].moveDistance++;
                //이동 거리가 다리의 길이와 같을 경우
                if (crossing[i].moveDistance == bridge_length)
                {
                    //passed 디큐(다리를 지나간 트럭들)에 트럭을 넣는다.
                    passed.push_back(crossing[i]);
                    //현재 다리의 무게에서 지나간 트럭의 무게를 빼준다.
                    bridge_weight -= crossing[i].weight;
                    //다리를 지나고 있는 crossing 디큐에서 지나간 트럭을 삭제한다.
                    crossing.pop_front();
                    //다시 첫번째 트럭을 검사할 수 있게 i를 --해준다.
                    i--; continue;
                }
            }
        }
        time++;
        //다리를 지나간 트럭의 수 == 트럭의 총 개수 일 경우 while문을 빠져나온다.
        if (passed.size() == truck_weights.size())break;
    }
    return time;
}

deque를 처음 사용하여 문제를 풀었다.

deque 는 index로 접근이 가능하면서도 pop_front, pop_back을 사용 가능하다.

장,단점을 숙지한 후 사용하면 편리할 것 같다.

 

Deque 링크

https://enum01.tistory.com/73?category=813879