[ 프로그래머스 ] 다리를 지나는 트럭

문제

---

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

풀이

---

모든 트럭이 일차선 다리를 건너려면 최소 몇 초가 걸리는지 알아내는 문제로, 한 번에 다리를 건널 수 있는 무게가 제한되어있다. 따라서 queue에 트럭의 무게와 시간을 넣고 시간을 측정하여 다음 트럭의 무게와 현재 다리에 있는 트럭 무게의 합이 다리의 하중보다 낮을 때 트럭을 queue에 추가하고, 현재 시간이 다리를 지나는데 필요한 트럭의 시간과 같을 때 queue에서 트럭을 pop하여 모든 트럭이 다리를 지났을 때의 시간을 return한다.

#include <string>
#include <vector>
#include <queue>

using namespace std;

int solution(int bridge_length, int weight, vector<int> truck_weights) 
{
    int answer = 0;

    queue<pair<int, int>> q;

    int cur_bridgeWeight = 0;
    int n = 0;
    int time = 0;

    while (1)
    {
        if (n != truck_weights.size())
        {
            if (weight - cur_bridgeWeight >= truck_weights[n])
            {
                q.push(make_pair(truck_weights[n], time + bridge_length));
                cur_bridgeWeight += truck_weights[n];
                n++;
                time++;
            }

            else
            {
                time++;
            }
        }

        else
        {
            time++;
        }

        pair<int, int> m = q.front();
        if (m.second <= time)
        {
            q.pop();
            cur_bridgeWeight -= m.first;
        }

        if (n == truck_weights.size() && q.empty())
        {
            break;
        }

    }

    answer = time + 1;

    return answer;
}