[프로그래머스] 이중우선순위큐

문제설명

이중 우선순위 큐는 다음 연산을 할 수 있는 자료구조를 말합니다.

명령어수신 탑(높이)

I 숫자	큐에 주어진 숫자를 삽입합니다.
D 1	큐에서 최댓값을 삭제합니다.
D -1	큐에서 최솟값을 삭제합니다.

이중 우선순위 큐가 할 연산 operations가 매개변수로 주어질 때, 모든 연산을 처리한 후 큐가 비어있으면 [0,0] 비어있지 않으면 [최댓값, 최솟값]을 return 하도록 solution 함수를 구현해주세요.

 

해결 방법

우선순위 큐를 사용해 해결할 수 있는 문제다.

 

명령에 따라 우선순위큐의 우선순위를 반전시켜준다.

 

D 1이 입력되었을 때는 내림차순을 기준으로 우선순위 큐를 만들어 저장된 값을 하나 제거한다.

 

D -1이 입력되었을 때는 오름차순을 기준으로 우선순위 큐를 만들어 저장된 값을 하나 제거한다.

 

JAVA

package programmers;

import java.util.*;
import java.io.*;

public class p_42628 {
	public static PriorityQueue del(int num, PriorityQueue<Integer> q) {
		PriorityQueue <Integer> n_q;
		if(num == 1) {
			n_q = new PriorityQueue<>(Collections.reverseOrder());
			while(!q.isEmpty()) {
				n_q.add(q.poll());
			}
		}
		else {
			n_q = new PriorityQueue<>();
			while(!q.isEmpty()) {
				n_q.add(q.poll());
			}
		}
		n_q.poll();
		return n_q;
	}
	public static int[] solution(String[] operations) {
		int[] answer = new int[2];
		PriorityQueue <Integer> q = new PriorityQueue<>();
		for(int i = 0; i < operations.length; i++) {
			char op = operations[i].charAt(0);
			int num = Integer.parseInt(operations[i].substring(2));
			if(op == 'I') q.add(num);
			else {
				if(q.isEmpty()) continue;
				else q = del(num, q);
			}
		}
		if(q.isEmpty()) {
			answer[0] = 0;
			answer[1] = 0;
		}
		else {
			PriorityQueue <Integer> maxq = new PriorityQueue<>(Collections.reverseOrder());
			PriorityQueue <Integer> minq = new PriorityQueue<>();
			while(!q.isEmpty()) {
				int num = q.poll();
				maxq.add(num);
				minq.add(num);
			}
			answer[0] = maxq.poll();
			answer[1] = minq.poll();
		}
		return answer;
		
	}
	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		String[] op = new String[n];
		for(int i = 0; i < n; i++) {
			op[i] = br.readLine();
		}
		int[] a = solution(op);
		System.out.println(a[0] + " " + a[1]);
	}
}