문제 URL : https://www.acmicpc.net/problem/2696
2696번: 중앙값 구하기
첫째 줄에 테스트 케이스의 개수 T(1 ≤ T ≤ 1,000)가 주어진다. 각 테스트 케이스의 첫째 줄에는 수열의 크기 M(1 ≤ M ≤ 9999, M은 홀수)이 주어지고, 그 다음 줄부터 이 수열의 원소가 차례대로 주
www.acmicpc.net
문제 접근법 : 중앙값을 구할려면 minheap과 maxheap을 이용하는 문제인데
사이즈가 같을땐 maxheap 넣어주면되고 다른경우 minheap 넣어줍니다.
그치만 maxheap에 갖고있는 데이터가 minheap보다 크면 minheap 안에있는 데이터를 swap하면 중앙값을 구할수있습닌다.
|
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
//By 콩순이냉장고
#include <bits/stdc++.h>
using namespace std;
vector<int> v;
int n;
void input() {
cin >> n;
v = vector<int>(n);
for (int i = 0; i < n; i++) {
cin >> v[i];
}
}
void solve() {
priority_queue<int> maxheap;
priority_queue<int, vector<int>, greater<int>> minheap;
vector<int> ans;
for (int i = 0; i < n; i++) {
if (maxheap.size() == minheap.size()) {
maxheap.push(v[i]);
}
else {
minheap.push(v[i]);
}
if (!minheap.empty() && maxheap.top() > minheap.top()) {
int temp = maxheap.top();
maxheap.pop();
maxheap.push(minheap.top());
minheap.pop();
minheap.push(temp);
}
if (i % 2 == 0)
ans.push_back(maxheap.top());
}
cout << ans.size() << "\n";
for (int i = 0; i < ans.size(); i++) {
cout << ans[i] << " ";
if ((i+1)%10 == 0)cout << "\n";
}
cout << "\n";
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(0);
cin.tie(0);
//freopen("input.txt", "r", stdin);
int test;
cin >> test;
while (test--) {
input();
solve();
}
}
|
cs |
궁금한점 혹은 모르는점 어떤 질문이든 댓글은 언제나 환영입니다.
'백준' 카테고리의 다른 글
| 백준 2342 Dance Dance Revolution (0) | 2021.11.02 |
|---|---|
| 백준 15644 구슬 탈출3 (0) | 2021.11.02 |
| 백준 11000 강의실 배정 (0) | 2021.10.30 |
| 백준 1715 카드 정렬하기 (0) | 2021.10.30 |
| 백준 8972 미친 아두이노 (1) | 2021.10.24 |
