-
-
Notifications
You must be signed in to change notification settings - Fork 297
/
Copy path295.cpp
112 lines (100 loc) · 2.84 KB
/
295.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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
__________________________________________________________________________________________________
sample 120 ms submission
static const int _ = [](){
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
return 0;
}();
class MedianFinder {
public:
/** initialize your data structure here. */
MedianFinder() {}
void addNum(int Num) {
if (LowerHalf.empty()) {
LowerHalf.push(Num);
return;
}
if (Num <= LowerHalf.top())
LowerHalf.push(Num);
else
UpperHalf.push(Num);
if (LowerHalf.size() > UpperHalf.size() + 1) {
UpperHalf.push(LowerHalf.top());
LowerHalf.pop();
} else if (UpperHalf.size() > LowerHalf.size() + 1) {
LowerHalf.push(UpperHalf.top());
UpperHalf.pop();
}
}
double findMedian() {
const auto Size = LowerHalf.size() + UpperHalf.size();
//assert(Size > 0 && "no number");
if (Size % 2 != 0) {
return LowerHalf.size() > UpperHalf.size() ?
LowerHalf.top() : UpperHalf.top();
} else
return (LowerHalf.top() + UpperHalf.top()) / 2.0;
}
private:
std::priority_queue<int> LowerHalf;
std::priority_queue<int, std::vector<int>, std::greater<int>> UpperHalf;
};
/**
* Your MedianFinder object will be instantiated and called as such:
* MedianFinder* obj = new MedianFinder();
* obj->addNum(num);
* double param_2 = obj->findMedian();
*/
__________________________________________________________________________________________________
sample 42244 kb submission
class MedianFinder {
public:
/** initialize your data structure here. */
vector<int> data;
MedianFinder() {
}
void addNum(int num) {
int start = 0;
int end = data.size() - 1;
if(data.empty())
{
data.push_back(num);
return;
}
while(start + 1 < end) {
int mid = (start + end) >> 1;
if(data[mid] <= num)
start = mid;
else
end = mid - 1;
}
int pos;
if(end >=0 && data[end] <= num)
pos = end;
else if(start >=0 && data[start] <= num)
pos = start;
else
pos = start - 1;
data.insert(data.begin() + pos + 1, num);
return;
}
double findMedian() {
if(data.size() % 2 == 0)
{
return ((double)(data[data.size()/2]) + (double)(data[data.size()/2-1])) / 2.0;
}
else
{
cout<<data.size()<<endl;
return data[data.size()/2];
}
}
};
/**
* Your MedianFinder object will be instantiated and called as such:
* MedianFinder obj = new MedianFinder();
* obj.addNum(num);
* double param_2 = obj.findMedian();
*/
__________________________________________________________________________________________________