-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathCookie Selection (Set).cpp
57 lines (53 loc) · 1.51 KB
/
Cookie Selection (Set).cpp
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
56
57
#include <iostream>
#include <map>
#include <string>
#include <algorithm>
#include <set>
#include <unordered_map>
#include <list>
#include <deque>
#include <cstdlib>
using namespace std;
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
multiset<int>set;
auto iter = set.begin();
string M;
while (cin >>M){
if (M == "#"){
if(set.size() % 2 == 0 ){ //set size even
cout<< *iter << endl;
auto iter2 = iter;
iter -- ;
set.erase(iter2);
}
else{ //set size odd
cout << *iter << endl;
auto iter2 = iter;
iter ++;
set.erase(iter2);
}}
else{
int K = stoi(M);
if (set.size() == 0){
set.insert(K);
iter = set.begin();
}
else if (set.size() % 2 ==0 ){ //if set size even
set.insert(K);
if (K < *iter){ //set size even and K smaller than median
iter -- ; }
}
else{ //if set size odd
set.insert(K);
if(K >= *iter){ //if set size odd and K bigger than median
iter ++;
}
}
// cout << "median "<< *iter <<endl;
}
}
return 0;
}