[JAVA] 백준 1927 최소 힙

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

1927번: 최소 힙

 

- 최소 힙을 이용하여 다음과 같은 연산을 지원하는 프로그램 

 

1. 배열에 자연수 x를 넣는다.

2. 배열에서 가장 작은 값을 출력하고, 그 값을 배열에서 제거 

 

- x가 자연수라면 배열에 x라는 값을 넣는(추가하는) 연산

- x가 0이라면 배열에서 가장 작은 값을 출력하고 그 값을 배열에서 제거

 

 

문제 풀이 

 

- 우선 순위가 낮은 숫자 순으로 우선 순위 큐를 선언하는 부분 제외하고 이전 문제와 동일하게 풀이 

 

 

정답 

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.PriorityQueue;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException
    {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        
        //우선순위가 낮은 순 = 최소힙
        PriorityQueue<Integer> pq = new PriorityQueue<>();
        int N = Integer.parseInt(br.readLine());
        
        StringTokenizer st;
        for(int i=0; i<N; i++)
        {
            st = new StringTokenizer(br.readLine());
            int value = Integer.parseInt(st.nextToken());
            
            //값 제거 
            if(value == 0)
            {
                if(pq.isEmpty())
                {
                    sb.append("0").append("\n");
                }
                else
                {
                    int ans = pq.poll();
                    sb.append(ans).append("\n");
                }
            }
            //값 추가 
            else
            {
                pq.add(value);
            }
        }
        
        System.out.println(sb);
    }
}