[백준 2024/01/20] 1927번 최소 힙

https://www.acmicpc.net/problem/1927

1927번: 최소 힙

 

1. 접근 방법

이번 문제는 11297번 최대 힙 을 구하는 문제에서 조금만 modify한다면, 바로 풀 수 있는 문제였다.

 

11297 문제의 아래 부분을

PriorityQueue<Integer> maxHeap = new PriorityQueue<>(Collections.reverseOrder());

 

아래와 같이 바꾸기만 하면 해결되었다.

PriorityQueue<Integer> minHeap = new PriorityQueue<>();

 

Java의 경우, minHeap의 경우만 지원하기 때문에,

Collections.reverseOrder()를 넣어서 maxHeap을 만들었었기 때문에

해당 부분만 지워주면 해결되었다.

 

2. 코드

import java.io.IOException;
import java.io.InputStreamReader;
import java.io.BufferedReader;
import java.util.PriorityQueue;
import java.util.Collections;
import java.util.Comparator;

public class b1927 {
    public static void main(String[] args) throws IOException {

        PriorityQueue<Integer> minHeap = new PriorityQueue<>();

        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
        String line = reader.readLine();
        int num = Integer.parseInt(line);
        for (int i=0; i<num; i++) {
            int new_num = Integer.parseInt(reader.readLine());

            if(new_num!=0){
                minHeap.add(new_num);
            }
            else{
                if(minHeap.isEmpty()){
                    System.out.println(0);
                }
                else {
                    System.out.println(minHeap.remove());
                }
            }


        }
    }

}