Skip to content

Commit c783023

Browse files
authored
Update sliding-window-median.cpp
1 parent a13a300 commit c783023

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

C++/sliding-window-median.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,9 @@
44
class Solution {
55
public:
66
vector<double> medianSlidingWindow(vector<int>& nums, int k) {
7-
multiset<int> min_bst(cbegin(nums), cbegin(nums) + k);
7+
multiset<double> min_bst(cbegin(nums), cbegin(nums) + k);
88
auto mid = next(cbegin(min_bst), k / 2);
9-
vector<double> result = {(double(*mid) + *prev(mid, 1 - k % 2)) / 2};
9+
vector<double> result = {((*mid) + *prev(mid, 1 - k % 2)) / 2};
1010
for (int i = k; i < size(nums); ++i) {
1111
min_bst.emplace(nums[i]);
1212
if (nums[i] < *mid) {
@@ -16,13 +16,12 @@ class Solution {
1616
++mid;
1717
}
1818
min_bst.erase(min_bst.lower_bound(nums[i - k]));
19-
result.emplace_back((double(*mid) + *prev(mid, 1 - k % 2)) / 2);
19+
result.emplace_back(((*mid) + *prev(mid, 1 - k % 2)) / 2);
2020
}
2121
return result;
2222
}
2323
};
2424

25-
2625
// Time: O(nlogk)
2726
// Space: O(k)
2827
class Solution2 {

0 commit comments

Comments
 (0)