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

• 기존에는 queue를 array리스트로 어떻게든 풀었는데 그걸 쓰지 않고서는 너무 복잡하게 코드가 짜여져서 시간이 많이 걸렸다.

아래는 queue를 이용한 해답이다.

import java.util.LinkedList;
import java.util.Queue;
import java.util.List;
import java.util.ArrayList;
 
public class Truck {
    int w;
    int d;
    
    public Truck(int w, int d) {
        this.w = w;
        this.d = d;
    }
}
 
class Solution {
    public int solution(int bridge_length, int weight, int[] truck_weights) {
        
        int wL = weight;
        int cnt = 0;
        Truck t = null;
        
        Queue<Truck> o = new LinkedList<Truck>();
        List<Truck> i = new ArrayList<Truck>();
 
        for (int l : truck_weights) {
            o.add(new Truck(l, bridge_length));
        }
 
        while (!(i.isEmpty()&&o.isEmpty())) {
            cnt++;
            
            if (!i.isEmpty() && i.get(0).d <= 0) {
                wL += i.get(0).w;
                i.remove(0);
            }
            
            if (!o.isEmpty() && wL - o.peek().w >= 0) {
                wL -= o.peek().w;
                i.add(o.poll());
            }
            
            for (int j = 0; j < i.size(); j++) {
                t = i.get(j);
                t.d--;
            }
        }
        return cnt;
    }
}

 for (int l : truck_weights) {
            o.add(new Truck(l, bridge_length));
        }

• 트럭을 전부 o에다가 저장을 해준다.
• 이때 트럭은 트럭의 무게, 다리의 길이의 매개변수를 가진다.

        while (!(i.isEmpty()&&o.isEmpty())) {
            cnt++;

• 다리 위 트럭(i)와 건너기 전 트럭(o)가 전부 빌 때까지 계속 실행을 하고 다리를 전부 건널 때의 시간을 측정하기 위해 cnt++해준다.

            if (!i.isEmpty() && i.get(0).d <= 0) {
                wL += i.get(0).w;
                i.remove(0);
            }

• 다리 위에 트럭이 있고 다리 위의 첫 번째 트럭의 남은 거리가 0이하 일 때
• 다리가 견딜 수 있는 하중(wL)을 첫 번째 트럭의 무게만큼 더해주고 다리 위 첫 번째 트럭을 제거해준다.

            if (!o.isEmpty() && wL - o.peek().w >= 0) {
                wL -= o.peek().w;
                i.add(o.poll());
            }

• 건너기 전 트럭(o)이 아직 있고 견딜 수 있는 하중(wL)이 건너려는 트럭의 무게를 더 견딜 수 있으면 견딜 수 있는 무게에 건너는 첫 트럭의 무게 만큼 빼주고 그 트럭은 다리 위에 있는 트럭(i)에서 뺴준다.

            for (int j = 0; j < i.size(); j++) {
                t = i.get(j);
                t.d--;
            }
        }
        return cnt;
    }
}

• 다리 위에 있는 트럭의 남아있는 거리를 시간이 가면 갈수록 하나씩 줄여준다. 마지막 리턴 값은 다 건넌 시간인 cnt가 반환이 된다. 
• 보통은 getter setter로 하지만 truck클래스에 필드의 접근제한자를 public이라고 지정해줬고 getter,setter를 따로 만들지 않았기 때문에 직접 남아있는 거리를 바꾸어주었다.