1. 程式人生 > 其它 >滑動視窗最大值java

滑動視窗最大值java

技術標籤:leetcode

給你一個整數陣列 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]
示例 3:

輸入:nums = [1,-1], k = 1
輸出:[1,-1]
示例 4:

輸入:nums = [9,11], k = 2
輸出:[11]
示例 5:

輸入:nums = [4,-2], k = 2
輸出:[4]

提示:

1 <= nums.length <= 105
-104 <= nums[i] <= 104
1 <= k <= nums.length

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/sliding-window-maximum

著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。

思路:太菜了不會資料結構只會暴力。暴力的就不貼出來了,貼一下官方的。

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        if(k==1)
            return nums;
        int n = nums.length;
        PriorityQueue<int[]> pq = new PriorityQueue<int[]>(new Comparator<
int[]>() { public int compare(int[] pair1, int[] pair2) { return pair1[0] != pair2[0] ? pair2[0] - pair1[0] : pair2[1] - pair1[1]; } }); //先只新增前k個 for (int i = 0; i < k; ++i) { pq.offer(new int[]{nums[i], i}); } int[] ans = new int[n - k + 1]; ans[0] = pq.peek()[0]; System.out.println(ans[0]); for (int i = k; i < n; ++i) { pq.offer(new int[]{nums[i], i}); //這一步很妙,只看最大值,如果你的最大值不在這個滑動視窗之中,那麼就丟擲 //不是最大值肯定就不在頂端,我不管你,有你沒你都不影響 while (pq.peek()[1] <= i - k) { pq.poll(); } ans[i - k + 1] = pq.peek()[0]; } return ans; } }