畢業舞會開幕!多人約會模擬器《魔物學園》即將登陸NS
阿新 • • 發佈:2021-12-16
給你一個整數陣列 nums,有一個大小為k的滑動視窗從陣列的最左側移動到陣列的最右側。你只可以看到在滑動視窗內的 k個數字。滑動視窗每次只向右移動一位。
返回滑動視窗中的最大值。
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/sliding-window-maximum
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。
優先佇列
import java.util.Comparator; import java.util.PriorityQueue; class Solution { public int[] maxSlidingWindow(int[] nums, int k) { 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]; } }); int[] ans = new int[n - k + 1]; for (int i = 0; i < n; ++i) { pq.offer(new int[]{nums[i], i}); if (i >= k - 1) { ans[i - k + 1] = pq.peek()[0]; while (!pq.isEmpty() && pq.peek()[1] <= i - k + 1) { pq.poll(); } } } return ans; } }
單調佇列
心之所向,素履以往 生如逆旅,一葦以航import java.util.Deque; import java.util.LinkedList; class Solution { public int[] maxSlidingWindow(int[] nums, int k) { int[] ret = new int[nums.length - k + 1]; Deque<Integer> queue = new LinkedList<>(); for (int i = 0; i < nums.length; ++i) { while (!queue.isEmpty() && nums[i] >= nums[queue.peekLast()]) { queue.pollLast(); } queue.offerLast(i); if (i >= k - 1) { ret[i - k + 1] = nums[queue.peekFirst()]; if (queue.peekFirst() == i - k + 1) { queue.pollFirst(); } } } return ret; } }