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

 

프로그래머스

점심시간 친구가 알고리즘 구현관련해서 질문을 해서 문제를 접하게 되었다.

질문했는데 내가 안풀고 대답하면, 이상한 대답 할 것이 분명하지 않은가?!!!?!?

 

1. 접근

보자마자 생각한것.

1. 경과시간이 왜 달라지는 것일까?
   • 아 bridge에 올라가는 weight 의 무게가 한정되어있구나. ⇒ 전역으로 weight을 합산해야지.
2. 제한조건은 어떻게 되는가??
   • bridge_length은 초와 직접적 상관있는것 ⇒ bridge애서 나오는 truck를 매초 확인해야겠다.

 

2.  풀이

• 초 마다 나오는 트럭이 있는지 확인해보자 ⇒ Deque 써야지
• 올라가있는 무게는 전역변수로 합산해서 관리하자.

import java.util.ArrayDeque;
import java.util.Deque;

class Solution {
    public int solution(int bridge_length, int weight, int[] truck_weights) {
        int answer = 0;
        Deque<Integer> states_bridge = new ArrayDeque<Integer>();
        int weigth_on_bridge = 0;
        
        for (int i = 0 ; i < bridge_length ; i++) states_bridge.add(0);
        
        
        int idx = 0;
        while(idx < truck_weights.length || !states_bridge.isEmpty()) {
            
            int truck = states_bridge.pop();
            if (truck != 0) weigth_on_bridge -= truck;
        
            if(idx < truck_weights.length){
                if (weigth_on_bridge + truck_weights[idx] <= weight) {
                    states_bridge.add(truck_weights[idx]);
                    weigth_on_bridge += truck_weights[idx];
                    idx++;
                } else {
                    states_bridge.add(0); 
                }
            }
            
            answer++;
        }
        
        return answer;
    }
}

 

이보다 더 좋은 코드, 좋은 접근이 많을거라고 생각한다. 그럼에도 시험에서는 빠르고 정확하게 푸는게 중요하니, 본인의 눈에 잘 들어오는 방법을 고안하는 것이 중요하다고 생각한다.