[프로그래머스/JAVA 자바] 다리를 지나는 트럭

 

프로그래머스

 

문제풀이

1. queue 활용 ( 선입선출 )

2. queue를 다리라고 생각하며 아래 상황을 고려

• 다리가 비어있으면
• 다리가 꽉차있으면
• 다리에 트럭이 있을때 ( 꽉 차지 않음 ) 무게가 초과하는지 안하는지

---

0초

---

1초

---

2초

---

3초

---

 

4초

---

5초

---

6초

나의풀이 ( 코드 )

import java.util.*;

class Solution {
    public int solution(int bridge_length, int weight, int[] truck_weights) {
        int answer = 0;
        int sumWeight = 0;
        int currentTruck = 0;

        Queue<Integer> queue = new LinkedList<>();

        for (int i : truck_weights) {
            currentTruck = i;

            while (true) {
                if (queue.isEmpty()) { // 다리에 트럭이 없으면 트럭 추가
                    queue.add(currentTruck);
                    sumWeight += currentTruck;
                    answer++;
                    break;
                } else if (queue.size() == bridge_length) { // 다리에 트럭이 꽉차있으면
                    sumWeight -= queue.poll();
                } else { // 트럭이 다리에 진입하기위해 무게 비교
                    if (sumWeight + currentTruck > weight) { // 무게 초과면
                        queue.add(0);
                        answer++;
                    }else{ // 트럭이 다리에 진입이 가능하면
                        queue.add(i);
                        sumWeight += i;
                        answer++;
                        break;
                    }
                }
            }


        }

        answer += bridge_length;
        return answer;
    }
}