문제
풀이
최소힙에서, pair<T, T>로 절대값 최소힙을 구현하는 코드이다. pair<절댓값, 실수값>으로 설정하여, 절대값이 제일 작은 것 중 실수값이 제일 작은 것, 즉, 절댓값이 같으면 음수가 top에 위치해있도록 하게 한다. 이렇게 구현하면 문제에서 원하는대로 구현이 되어 쉽게 top의 second를 출력하면 되는 문제이다. pair를 이용한 절대값 최소힙을 알아두면 좋을 것 같다.
소스 코드
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cstdlib>
using namespace std;
typedef pair <int, int> pi;
int main() {
int N, num, i;
pi temp;
priority_queue<pi, vector<pi>, greater<pi> > q;
scanf("%d", &N);
for(i = 0; i < N; i++) {
scanf("%d", &num);
if (num == 0) {
if (q.empty()) printf("0\n");
else {
temp = q.top();
printf("%d\n", temp.second);
q.pop();
}
}
else {
q.push(make_pair(abs(num), num));
}
}
return 0;
}
