11279번 : 최대 힙

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

11279번: 최대 힙

 

 

문제

 

 

코드

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();
        int n = Integer.parseInt(br.readLine());
        PriorityQueue<Integer> pq = new PriorityQueue<Integer>(Collections.reverseOrder());
        
        for(int i=0;i<n;i++) {
            int input = Integer.parseInt(br.readLine());
            if(input==0) {
                Integer p = pq.poll();
                if(p == null) sb.append(0 + "\n");
                else sb.append(p + "\n");
            } else {
                pq.add(input);
            }
        }
        bw.write(sb+""); bw.flush();

    }
}

 

 

해결방법

자바의 java.util.PriorityQueue 클래스를 사용하면 쉽게 해결할 수 있는 문제이다. 

우선순위 큐를 생성할 때, Collections.reverseOrder()를 인자로 주면 우선순위가 큰 값이 되어서 최대 힙을 만들 수 있다.