아찌방 하우스

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

풀이 - 나의 생각

큐를 트럭과 다리를 담당하도록

2개를 선언했다.

1. Truck Queue

우선 Queue에 truck을 다 집어 넣는다.

2. Bridge Queue

우선 Queue에 다리 길이 만큼 0을 다 집어 넣는다.

이때 0은 truck이 올라가 있지 않다는 것을 의미한다.

3. 반복문 돌리기

while문으로 Bridge Queue가 비어 있지 않는 경우 반복문을 돌린다.

다리에 올라가 있는 트럭의 무게를 체크하면서

다음 트럭을 올려도 되는지, 안 되는지 파악한다.

코드

import java.util.*;
class Solution {
    public int solution(int bridge_length, int weight, int[] truck_weights) {
        int time = 0;
        int current_weight = 0;

        Queue<Integer> bridge = new LinkedList<>();
        Queue<Integer> trucks = new LinkedList<>();

        for(int truck : truck_weights){
            trucks.offer(truck);
        }

        for(int i = 0; i < bridge_length; i++){
            bridge.offer(0);
        }
        
        while(!bridge.isEmpty()){
            int passed_truck = bridge.poll();
            current_weight -= passed_truck;
            
            if(!trucks.isEmpty()){
                int next_truck = trucks.peek();
                
                if(current_weight + next_truck <= weight){
                    bridge.offer(trucks.poll());
                    current_weight += next_truck;
                }else{
                    bridge.offer(0);
                }
            }
            
            time++;
        }
        
        return time;
    }
}

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

풀이 - 나의 생각

PriorityQueue를 2개 쓰면 된다.

(한개는 최솟값, 한개는 최댓값을 위한)

PriorityQueue는 기본적으로는 오름차순으로 정렬된다.

내림차순을 원한다면

PriorityQueue<Integer> pq = new PriorityQueue<>((a, b) -> Integer.compare(b, a));

이런식으로 해줘야 한다.

코드

import java.util.*;

class Solution {
    public int[] solution(String[] operations) {
        PriorityQueue<Integer> maxQueue = new PriorityQueue<>((a, b) -> Integer.compare(b, a));
        PriorityQueue<Integer> minQueue = new PriorityQueue<>();
        
        for(String op : operations){
            String[] tmp = op.split(" ");
            int num = Integer.parseInt(tmp[1]);
            if(tmp[0].equals("I")){
                maxQueue.offer(num);
                minQueue.offer(num);
            }else{
                if(num == 1 && !maxQueue.isEmpty()){
                    int max = maxQueue.poll();
                    minQueue.remove(max);
                }else if(!minQueue.isEmpty()){
                    int min = minQueue.poll();
                    maxQueue.remove(min);
                }
            }
        }
        
        int min = 0, max = 0;
        
        if(!maxQueue.isEmpty()){
            min = minQueue.poll();
            max = maxQueue.poll();
        }
        
        return new int[] {max,min};
    }
}