中位數是有序序列最中間的那個數。如果序列的大小是偶數，則沒有最中間的數；此時中位數是最中間的兩個數的平均數。

例如：

[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);} }

總結

以上是生活随笔為你收集整理的leetcode 480. 滑动窗口中位数（堆+滑动窗口）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。