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

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

코딩테스트 연습 - 다리를 지나는 트럭 | 프로그래머스

풀이 :

cand의 second에는 +1씩 증가하도록 하여 길이와 같아진다면 erase 하는 방식으로 해결하였다.

 

코드 ( C++ )

 

	#include <iostream>
	#include <queue>
	#include <vector>
	using namespace std;
	int solution(int bridge_length, int weight, vector<int> truck_weights) {
	int answer = 0;
	queue<int> list;
	for (int i = 0; i < truck_weights.size(); ++i)
	list.push(truck_weights[i]);
	int time = 1;
	vector<pair<int, int>> cand;
	int sum = 0;
	while(1)
	{
	if (!list.empty() && sum + list.front() <= weight) {
	cand.push_back({ list.front(),0 });
	sum += list.front();
	list.pop();
	}
	for (int i = 0; i < cand.size(); ++i)
	{
	cand[i].second++;
	}
	time++;
	if (cand[0].second == bridge_length) {
	sum -= cand[0].first;
	cand.erase(cand.begin());
	}
	if (list.empty() && cand.empty())
	break;
	}
	return time;
	}

view raw .cpp hosted with ❤ by GitHub