中位數是有序序列最中間的那個數。如果序列的大小是偶數,則沒有最中間的數;此時中位數是最中間的兩個數的平均數。
例如:
[2,3,4],中位數是 3
[2,3],中位數是 (2 + 3) / 2 = 2.5
給你一個數組 nums,有一個大小為 k 的窗口從最左端滑動到最右端。窗口中有 k 個數,每次窗口向右移動 1 位。你的任務是找出每次窗口移動后得到的新窗口中元素的中位數,并輸出由它們組成的數組。
示例:
給出 nums = [1,3,-1,-3,5,3,6,7],以及 k = 3。
窗口位置 中位數
[1 3 -1] -3 5 3 6 7 1
1 [3 -1 -3] 5 3 6 7 -1
1 3 [-1 -3 5] 3 6 7 -1
1 3 -1 [-3 5 3] 6 7 3
1 3 -1 -3 [5 3 6] 7 5
1 3 -1 -3 5 [3 6 7] 6
因此,返回該滑動窗口的中位數數組 [1,-1,-1,3,5,6]。
解題思路
利用大根堆+小根堆維護窗口內的元素,兩個堆各維持窗口內的一半元素,而中位數則是根據k的奇偶,從max.peek()和min.peek()當中產生,如果k是奇數,則大根堆比小根堆個數多1,且堆頂就是中位數
代碼
class Solution {PriorityQueue<Double> max=new PriorityQueue<>((o1, o2) -> (int) (o2-o1));PriorityQueue<Double> min=new PriorityQueue<>();public double[] medianSlidingWindow(int[] nums, int k) {int r=0,cnt=0;double[] res=new double[nums.length+1-k];for(int l=0;l+k-1<nums.length;l++){while (r<nums.length&&cnt<k){add(nums, r);r++;cnt++;}if(max.size()==min.size()){res[l]=(max.peek()+min.peek())*0.5;}else res[l]=max.peek();remove(nums, l);cnt--;}return res;}public void add(int[] nums, int k) {max.add((double) nums[k]);min.add(max.poll());if(min.size()>max.size()){max.add(min.poll());}}public void remove(int[] nums, int k) {double temp=(double)nums[k];if(max.peek()>=temp)max.remove(temp);else min.remove(temp);}
}