본문 바로가기

Programmers

[프로그래머스][JAVA]Lv. 2 - 다리를 지나는 트럭

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

큐를 이용. 먼저 다리의 길이만큼 0을 넣어주고 다리 위 전체 무게인 totalWeight + 다음 대기 트럭이 weight 보다 작거나 같다면 큐에 넣어주고 weight 보다 크다면 0을 넣어준 다음 큐를 poll 해주는 것을 반복한다.

이 과정에서 다리의 길이만큼 0이 제거되고 뒤쪽으로 0이 추가되면서 시간은 1초씩 증가하게 된다. 그리고 다시 다리 길이를 정답에 더해주어 건너는데 걸린 시간 + 다리길이로 정답을 구할 수 있었다.

import java.util.LinkedList;
import java.util.Queue;
class Solution {
    public int solution(int bridge_length, int weight, int[] truck_weights) {
        int answer = 0;
        int totalWeight = 0;
        Queue<Integer> queue = new LinkedList<>();

        for (int i = 0; i < bridge_length; i++) {
            queue.add(0);
        }

        int index = 0;
        while (!queue.isEmpty()) {
            totalWeight -= queue.poll();
            if (index == truck_weights.length) break;
            if (totalWeight + truck_weights[index] <= weight) {
                queue.offer(truck_weights[index]);
                totalWeight += truck_weights[index];
                index++;
            } else {
                queue.offer(0);
            }
            answer++;
        }

        return answer + bridge_length;
    }
}