[1655] 가운데를 말해요
BOJ/Data Structure2018. 7. 22. 12:03
### Priority Queue ###
[1655] 가운데를 말해요 : http://boj.kr/1655
<소스코드>
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 | #include <cstdio> #include <queue> using namespace std; int main() { int n, k; scanf("%d", &n); priority_queue<int> minQ, maxQ; scanf("%d", &k); printf("%d\n", k); maxQ.push(k); for(int i=1; i<n; i++) { scanf("%d", &k); if(i&1) { if(maxQ.top() < k) minQ.push(-k); else { maxQ.push(k); minQ.push(-maxQ.top()); maxQ.pop(); } } else { if(-minQ.top() > k) maxQ.push(k); else { minQ.push(-k); maxQ.push(-minQ.top()); minQ.pop(); } } printf("%d\n", maxQ.top()); } return 0; } | cs |
>> max Heap 과 min Heap 을 이용하여, 그때그때마다 중앙값을 알 수 있다.
- input : 1 5 2 10 -99 7 5 순이라면
1) max Heap : 1
min Heap : empty
2) max Heap : 1
min Heap : 5
3) max Heap : 1 2
min Heap : 5
4) max Heap : 1 2
min Heap : 5 10
5) max Heap : -99 1 2
min Heap : 5 10
6) max Heap : -99 1 2
min Heap : 5 7 10
이런식으로 진행이 된다. 그때그때 마다의 중앙값은 max Heap 의 루트가 된다.
'BOJ > Data Structure' 카테고리의 다른 글
[5639] 이진 검색 트리 (0) | 2018.07.22 |
---|---|
[2529] 부등호 (0) | 2018.07.22 |