Lintcode360 Sliding Window Median solution 题解

【题目描述】

Given an array of n integer, and a moving window(size k), move the window at each iteration from the start of the array, find the median of the element inside the window at each moving. (If there are even numbers in the array, return the N/2-th number after sorting the element in the window. )

给定一个包含 n 个整数的数组，和一个大小为k的滑动窗口,从左到右在数组中滑动这个窗口，找到数组中每个窗口内的中位数。(如果数组个数是偶数，则在该窗口排序数字后，返回第 N/2 个数字。)

【题目链接】

www.lintcode.com/en/problem/sliding-window-median/

【题目解析】

这道题和Data Stream Median类似，也是寻找动态数组的media，因此也可以用maxheap＋minheap来解。随着窗口的移动，每次先加入一个新元素，再删去一个旧元素，再使两个heap中元素数量平衡即可。

用一个最大堆来记录较小一半的元素，一个最小堆来记录较大一半的元素。

先初始化最初的窗口里的k个元素。将元素加入最大堆，若为奇数次，则比较最大堆堆顶和最小堆堆顶元素大小，若最大堆堆顶元素大于和最小堆堆顶元素，则交换两个堆顶元素，若为偶数次，则将最大堆堆顶元素加入最小堆。

然后开始移动窗口，先将新元素加入，若比原median小，则加入最大堆，反之则加入最小堆。然后删除窗口最前面的一个元素。

然后调整最大堆和最小堆的大小。根据k值可以分两种情况讨论：

1）若k为偶数，则需要最大堆中元素和最小堆中元素数量相等

2）若k为奇数，则需要最大堆中元素比最小堆中元素多1个

此时最大堆的堆顶元素即为此时窗口元素的median

【参考答案】

www.jiuzhang.com/solutions/sliding-window-median/