1、题目来源
2、题目描述
给你一个整数数组 nums
,有一个大小为 k
的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k
个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
示例 1:
输入:nums = [1,3,-1,-3,5,3,6,7], k = 3 输出:[3,3,5,5,6,7] 解释: 滑动窗口的位置 最大值 --------------- ----- [1 3 -1] -3 5 3 6 7 3 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 5 1 3 -1 -3 [5 3 6] 7 6 1 3 -1 -3 5 [3 6 7] 7
示例 2:
输入:nums = [1], k = 1 输出:[1]
提示:
1 <= nums.length <= 105
-104 <= nums[i] <= 104
1 <= k <= nums.length
3、题解分享
// 方法一
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
// 思路:优先队列 + 将k个数放入大根堆中,通过堆顶查看最大值
int n = nums.length;
int ansLen = n - k + 1;
int[] ans = new int[ansLen];
int index = 0;
PriorityQueue<int[]> pq = new PriorityQueue<>(new Comparator<int[]>(){
public int compare(int[] a,int[] b){
// 构造大根堆
if(a[0] == b[0]){
return b[1] - a[1];
}
return b[0] - a[0];
}
});
for(int i = 0;i<k;++i){
pq.offer(new int[]{nums[i],i});
}
ans[index++] = pq.peek()[0];
for(int i = k;i<n;++i){
pq.offer(new int[]{nums[i],i});
while(pq.peek()[1] <= i - k){// 将窗口外的数弹出
pq.poll();
}
ans[index++] = pq.peek()[0];
}
return ans;
}
}
// 方法二
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
//思路:单调队列 + 如果有两个下标 i < j在窗口内,并且nums[i] < nums[j],则nums[i]永远不会是滑动窗口内的最大值
//将窗口内的数按照从大到小排列在单调队列中(存储对应的下标)
int n = nums.length;
int ansLen = n - k + 1;
int[] ans = new int[ansLen];
int index = 0;
Deque<Integer> q = new LinkedList<>();
for(int i = 0;i<k;++i){
while(!q.isEmpty() && nums[i] >= nums[q.peekLast()]){
q.pollLast();
}
q.offerLast(i);
}
ans[index++] = nums[q.peekFirst()];
for(int i =k;i<n;++i){
while(!q.isEmpty() && nums[i] > nums[q.peekLast()]){
q.pollLast();
}
q.offerLast(i);
while(!q.isEmpty() && q.peekFirst() <= i - k){
q.pollFirst();
}
ans[index++] = nums[q.peekFirst()];
}
return ans;
}
}
// 方法三
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
//思路:分块 + 预处理,将数组nums按k的大小进行分块,要求滑动窗口nums[i] 到 nums[i+k−1]的最大值,就会有两种情况:
// (1)如果i是k的倍数,那么nums[i] 到 nums[i+k−1]就在同一个窗口,只要预处理得到这个分组的最大值即可
// (2)如果i不是k的倍数,那么nums[i] 到 nums[i+k−1]会横跨两个分组,那么只要比较 前分组的后缀最大值 和 后分组的前缀的最大值,取较大即可
int n = nums.length;
int[] ans = new int[n - k + 1];
// pre和suf都存储i所在分组的最大值
int[] prefixMax = new int[n];
int[] suffixMax = new int[n];
// 求前缀
for(int i = 0;i<n;++i){
if(i % k == 0){
prefixMax[i] = nums[i];
}else{
prefixMax[i] = Math.max(nums[i],prefixMax[i-1]);
}
}
// 求后缀
for(int i = n-1;i>=0;--i){
if(i == n-1 || (i+1) % k == 0){
suffixMax[i] = nums[i];
}else{
suffixMax[i] = Math.max(suffixMax[i+1],nums[i]);
}
}
for(int i = 0;i<n - k +1;++i){
ans[i] = Math.max(prefixMax[i+k-1],suffixMax[i]);
}
return ans;
}
}