[프로그래머스]스택/큐 : 다리를 지나는 트럭 (level 2)(c++)

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

프로그래머스

 

 

#include <string>
#include <vector>
#include <queue>
 
using namespace std;
 
int solution(int bridge_length, int weight, vector<int> truck_weights) {
 
    queue<int> q;
    vector<int> pos(truck_weights.size());
    int sum_weight = 0, idx = 0, tm = 0;
 
    while (!q.empty() || idx < truck_weights.size())
    {
        tm++;
        if (!q.empty())
        {
            // 위치 이동
            for (int i = q.front(); i <= q.back(); i++)
            {
                pos[i]++;
            }
 
            // 다리 바깥으로 나올때
            if (pos[q.front()] >= bridge_length)
            {
                sum_weight -= truck_weights[q.front()];
                q.pop();
            }
        }
 
        // 다리위 트럭 무게 + 이번 트럭 무게 <= weight --> push
        if (idx < truck_weights.size() && sum_weight + truck_weights[idx] <= weight)
        {
            sum_weight += truck_weights[idx];
            q.push(idx);
            idx++;
        }
    }
 
    return tm;
}

 

 

기능개발 문제와 비슷하다고 생각했지만,

기능개발은 시간만 카운팅하면 되고

이번 문제는 시간, 위치까지 카운팅 해야했다.

 

전체 트럭개수 만큼만 for문을 돌려보려했지만 실패했다 ㅜㅜ