JZ-064-滑動視窗的最大值

雄獅虎豹發表於2022-02-24

滑動視窗的最大值

題目描述

給定一個陣列和滑動視窗的大小，找出所有滑動視窗裡數值的最大值。例如，如果輸入陣列{2,3,4,2,6,2,5,1}及滑動視窗的大小3，那麼一共存在6個滑動視窗，他們的最大值分別為{4,4,6,6,6,5}； 針對陣列{2,3,4,2,6,2,5,1}的滑動視窗有以下6個： {[2,3,4],2,6,2,5,1}， {2,[3,4,2],6,2,5,1}，{2,3,[4,2,6],2,5,1}， {2,3,4,[2,6,2],5,1}， {2,3,4,2,[6,2,5],1}， {2,3,4,2,6,[2,5,1]}。

  • 視窗大於陣列長度的時候，返回空。

題目連結: 滑動視窗的最大值

程式碼

import java.util.ArrayList;
import java.util.PriorityQueue;

/**
 * 標題：滑動視窗的最大值
 * 題目描述
 * 給定一個陣列和滑動視窗的大小，找出所有滑動視窗裡數值的最大值。例如，如果輸入陣列{2,3,4,2,6,2,5,1}及滑動視窗的大小3，那麼一共存在6個滑動視窗，
 * 他們的最大值分別為{4,4,6,6,6,5}； 針對陣列{2,3,4,2,6,2,5,1}的滑動視窗有以下6個： {[2,3,4],2,6,2,5,1}， {2,[3,4,2],6,2,5,1}，
 * {2,3,[4,2,6],2,5,1}， {2,3,4,[2,6,2],5,1}， {2,3,4,2,[6,2,5],1}， {2,3,4,2,6,[2,5,1]}。
 * 視窗大於陣列長度的時候，返回空
 * 題目連結：
 * https://www.nowcoder.com/practice/1624bc35a45c42c0bc17d17fa0cba788?tpId=13&&tqId=11217&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking
 */
public class Jz64 {

    public ArrayList<Integer> maxInWindows(int[] num, int size) {
        ArrayList<Integer> result = new ArrayList<>();
        if (size > num.length || size < 1) {
            return result;
        }

        PriorityQueue<Integer> heap = new PriorityQueue<>((o1, o2) -> o2 - o1); // 大頂堆
        for (int i = 0; i < size; i++) {
            heap.add(num[i]);
        }
        result.add(heap.peek());
        for (int i = 0, j = i + size; j < num.length; i++, j++) { // 維護一個大小為 size 的大頂堆
            heap.remove(num[i]);
            heap.add(num[j]);
            result.add(heap.peek());
        }
        return result;
    }

    public static void main(String[] args) {
    
    }
}
【每日寄語】 撕掉固有標籤，發現更讚的自己！

相關文章