프로그래머스 Lv3_이중우선순위큐_Java

https://school.programmers.co.kr/learn/courses/30/lessons/42628?language=java

프로그래머스

 

import java.util.*;

class Solution {
    public int[] solution(String[] operations) {
        int[] answer = {};
        PriorityQueue<Integer> pq = new PriorityQueue<>(); // 오름차순
        PriorityQueue<Integer> reversePq = new PriorityQueue<>(Collections.reverseOrder()); // 내림차순
        
        for(String n : operations) {
            String[] arr = n.split(" ");
            String check = arr[0];
            int value = Integer.parseInt(arr[1]);
            
            if(check.equals("I")) {  
                pq.add(value);
                reversePq.add(value);
            } else if (check.equals("D") && value == -1) {
                if(!pq.isEmpty()) {
                    int a = pq.poll();
                    reversePq.remove(a);
                }
            } else if (check.equals("D") && value == 1) {
                if(!reversePq.isEmpty()) {
                    int maxValue = reversePq.poll();
                    pq.remove(maxValue);
                }
            }
        }
        
        if(pq.isEmpty()) {
            answer = new int[]{0, 0};
        } else {
            answer = new int[]{reversePq.poll(), pq.poll()};
        }
        
        return answer;
    }
}

 

풀이

- 우선순위 큐 두개로 내림차순 오름차순을 만들어 각각 데이터를 저장한다

- 저장 후 문자열을 비교하여 최솟값과 최댓값을 빼준다

- 모든 것을 탐색하고 queue에 데이터가 남아있으면 최소 최댓값을 출력 비어 있으면 0 0 리턴

 

회고

- queue 두개로 queue와 stack을 둘 다 사용할 수 있냐고 묻는 문제 같았다