中位数是有序序列最中间的那个数。如果序列的大小是偶数,则没有最中间的数;此时中位数是最中间的两个数的平均数。
例如:
[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);}
}