如题所示
给定一个数组 nums 和滑动窗口的大小 k,请找出所有滑动窗口里的最大值。
示例:
输入: 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
package Pta;
import java.util.Arrays;
import java.util.Comparator;
import java.util.PriorityQueue;
public class Test {
public static void main(String[] args) {
Solution99 s=new Solution99();
System.out.println(Arrays.toString(s.maxSlidingWindow(new int[]{1,3,-1,-3,5,3,6,7},3)));
}
}
class Solution99 {
public int[] maxSlidingWindow(int[] nums, int k) {
PriorityQueue queue=new PriorityQueue(new Comparator<Integer>() {
@Override
public int compare(Integer a, Integer b) {
return b-a;
}
});
int[] res=new int[nums.length-k+1];
for(int i=0;i<k;i++){
queue.offer(nums[i]);
}
res[0]=(int) queue.peek();
//System.out.println(queue);
for(int i=1;i<nums.length-k+1;i++){//i=5
queue.remove(nums[i-1]);
queue.offer(nums[i+k-1]);
res[i]= (int) queue.peek();
}
return res;
}
}