99클럽 코테 스터디 9일차 TIL + 최소 힙 (java)

 

💡 문제

• 최소 힙
• https://www.acmicpc.net/problem/1927

 

권장 시간

• 1시간 30분

 

소요 시간

• 30분

 

풀이 코드

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

public class BOJ1927 {
    public void solution() throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

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

        int N = Integer.parseInt(br.readLine());
        for (int i = 0; i < N; i++) {
            int num = Integer.parseInt(br.readLine());
            if (num == 0) { // 가장 작은 값을 출력하고, 그 값을 배열에서 제거하는 경우
                if (pq.isEmpty()) {
                    System.out.println(0);
                } else {
                    System.out.println(pq.poll());
                }
            } else {
                pq.add(num);
            }
        }
    }

    public static void main(String[] args) throws Exception {
        new BOJ1927().solution();
    }
}

주요사항

우선순위 큐를 활용하면 금방 풀 수 있는 문제!

우선순위 큐는 minHeap과 같은 우선순위로 정렬된다.

maxHeap은 compare 함수를 Override 해서 반대로 세팅해주면 되겠죵?