Baekjoon Review

[Silver 2] 1927번 최소 힙

hanseongbugi 2023. 11. 27. 15:45

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

 

1927번: 최소 힙

첫째 줄에 연산의 개수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 N개의 줄에는 연산에 대한 정보를 나타내는 정수 x가 주어진다. 만약 x가 자연수라면 배열에 x라는 값을 넣는(추가하는) 연산이고, x가 0

www.acmicpc.net

 

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

삽입할 수가 0이면 큐에서 빼서 출력하고 아니면 삽입하면 최소 힙을 구현할 수 있다.

 

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
using namespace std;

int N;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	cin >> N;
	priority_queue<int, vector<int>, greater<int>> heap;

	int value;
	for (int i = 0; i < N; i++) {
		cin >> value;
		if (value == 0) {
			if (heap.empty()) cout << 0 << '\n';
			else {
				cout << heap.top() << '\n';
				heap.pop();
			}
		}
		else {
			heap.push(value);
		}
	}
}