다리를 지나는 트럭 (Java)

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

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

 

프로그래머스 코딩테스트 고득점 Kit 다리를 지나는 트럭 문제 풀이이다.

bridge_length가 다리의 길이이자 한 트럭이 지나가는데 걸리는 시간이라는 점을 고려하여 풀어야 한다.

 

일차선 다리는 트럭이 앞에서 뒤로 지나가므로 큐를 일차선 다리라고 생각하고 풀었다.

 

일단 큐(다리)에 있는 전체 하중을 구해 이번에 들어갈 하중을 더했을 때 수용 가능하면 넣어주었고, 커서 못 들어가는 경우는 다리가 비어있다는 의미(다리가 비어있어도 지나는데 시간이 걸리므로)에서 0으로 채워주었다. 다리가 꽉 차있으면 마지막까지 이동했다는 의미이므로 마지막 요소를 제거해주었다.

 

import java.util.*;

class Solution {
    public int solution(int bridge_length, int weight, int[] truck_weights) {
        int i = 0;
        int answer = 0; // 마지막 트럭 제외 모든 트럭이 다리를 통과하는데 걸린 시간
        Queue<Integer> bridge = new LinkedList<>(); // 일차선 다리
        
        while (true) {
            int turn = truck_weights[i];
            if (i == truck_weights.length) {
                break;
            }
            if (bridge.size() == bridge_length) {
                bridge.poll();
            } else if (getBridgeSum(bridge) + turn > weight) {
                bridge.add(0);
                answer++;
            } else {
                bridge.add(turn);
                answer++;
                i++;
            }
        }
        return answer + bridge_length;
    }
    
    public int getBridgeSum(Queue<Integer> que) {
        Iterator iter = que.iterator();
        int sum = 0;
        while (iter.hasNext()) {
            sum = sum + (Integer) iter.next(); 
        }
        return sum;
    }
}

 

입출력 예시에 있는 케이스 순서로 반복이 끝났을 때 변수 값을 출력해보면 아래와 같다.

bridge (큐)	answer
7	1
7, 0	2
0, 4	3
4, 5	4
5, 0	5
0, 6	6