개요

다리를 지나는 트럭

• 문제 설명
  트럭 여러 대가 강을 가로지르는 일차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 다리에는 트럭이 최대 bridge_length대 올라갈 수 있으며, 다리는 weight 이하까지의 무게를 견딜 수 있습니다. 단, 다리에 완전히 오르지 않은 트럭의 무게는 무시합니다.

  예를 들어, 트럭 2대가 올라갈 수 있고 무게를 10kg까지 견디는 다리가 있습니다. 무게가 [7, 4, 5, 6]kg인 트럭이 순서대로 최단 시간 안에 다리를 건너려면 다음과 같이 건너야 합니다.

  경과 시간/ 다리를 지난 트럭/ 다리를 건너는 트럭/ 대기 트럭
  0 / [] / [] / [7,4,5,6]
  1~2 / [] / [7] / [4,5,6]
  3 / [7] / [4] / [5,6]
  4 / [7] / [4,5] / [6]
  5 / [7,4] / [5] / [6]
  6~7 / [7,4,5] / [6] / []
  8 / [7,4,5,6] / [] / []
  따라서, 모든 트럭이 다리를 지나려면 최소 8초가 걸립니다.

  solution 함수의 매개변수로 다리에 올라갈 수 있는 트럭 수 bridge_length, 다리가 견딜 수 있는 무게 weight, 트럭 별 무게 truck_weights가 주어집니다. 이때 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 return 하도록 solution 함수를 완성하세요.

접근 방식

1. pair<int,int>형을 자료형으로 가지는 큐 road를 선언했다.
   first값엔 각 트럭의 무게, second값엔 현재 트럭이 도로를 벗어날때 시간이다.

2. 현재 시간을 curTime으로 정하고 도로의 현재 무게를 curWeight으로 선언했다.

3. 트럭 벡터를 순회하며, 각 트럭의 차례가 될때 curTime을 1씩 증가시킨다.
   각 순회에선 세가지 케이스가 있다.

   1. 도로의 차 숫자가 가득찬 경우
      -> 도로의 front트럭을 pop, 해당 트럭의 무게만큼 curWeight에서 제거,
      curTime값을 front트럭의 도로탈출 예정시간인 front().second값으로 변경
   2. 도로의 무게가 꽉찬 경우
      -> 위와 동일하게 처리
      주의할점은 현재 무게가 들어갈 수 있을때까지 위 과정을 반복해야한다.
   3. 나머지
      -> 현재 curTime에 bridge의 길이만큼 더한 후, 큐에 저장한다.

4. 처음엔 각 초마다 큐의 front트럭이 나올 수 있는지 체크했지만,
   생각해보니 그럴 필요가 없었다.
   각 트럭의 차례에서 들어갈수 있다면 들어가고 못 들어가면,
   큐의 front트럭이 나올 시간으로 시간을 설정하고 빼버리면 되는 것이였다.

전체 코드

#include <string>
#include <vector>
#include<queue>
#include<algorithm>
#include<iostream>

using namespace std;

int solution(int bridge_length, int weight, vector<int> truck_weights) {
    int answer = 0;
    //first는 무게 second는 road체류시간
    queue<pair<int,int>> road;
    int curWeight=0,curTime=0;
    
    for(auto iter=truck_weights.begin();iter!= truck_weights.end();++iter){
        curTime++;
        //도로가 꽉찬 경우라면
        if(road.size()==bridge_length){
            //현재 시간을 맨앞차량이 나올예정인 시간으로 설정
            curTime=road.front().second;
            curWeight-=road.front().first;
            road.pop();
        }
        //도로의 무게가 꽉찬 경우라면
        if(curWeight+(*iter)> weight) {
            while(curWeight+(*iter) > weight ){
                //위와 동일하게 처리
                curTime=max(curTime,road.front().second);
                curWeight-=road.front().first;
                road.pop();
            }
        }
        curWeight+= (*iter);
        road.push({(*iter),curTime+bridge_length});
        
    }

    while(!road.empty()){
        curTime=max(curTime,road.front().second);
        road.pop();
    }
    return answer=curTime;
}

문풀후생

이 구조를 떠올리는데 많은 시간을 투자했고 구현을 했다.

하지만 테스트케이스 1부터 계속 틀려서 뭔가하고 분석해봤더니 두 가지 실수를 했다.
일단 각 트럭이 들어갈 수 있는 무게까지 큐에서 트럭을 빼야한다.

front에 해당되는 트럭 하나만 빼서, 못 들어가는 상황인데 들어간 처리가 된것이다.

두 번째는 케이스를 나눌 때,

if(road.size()==bridge_length){
}
//도로의 무게가 꽉찬 경우라면
else if(curWeight+(*iter)> weight) {

이런 식으로 둘 중 하나만 처리했다.

하지만 도로가 가득 찼을 때, front()트럭 하나만 빼고 무게 고려안하고
현재 차례의 트럭이 들어가버리면 안된다.

예를들어 도로에 트럭 최대 3대가 올라갈 수 있고, 최대 무게가 10일 경우,
{2,3,4} 이렇게 올라가 있고, 무게 9짜리 트럭이 들어갈 경우,
2짜리 트럭 하나 빼고 {3,4,9} 이럴 순 없다.
9가 들어갈 수 있도록 2,3,4를 다 빼야한다 .

따라서 else if가 아니라 둘다 if를 작성해서 두가지 다 예외처리를 해야한다.