算法代码

滑动窗口最大值

2020-09-10  本文已影响0人  windUtterance

题目描述
给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回滑动窗口中的最大值。

示例

image.png

Java代码

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        if(nums.length == 0) return new int[0];

        int len = nums.length;
        int[] arr = new int[len - k + 1];
        /*首先维护一个窗口的最大值
        如果新窗口的范围超过了这个最大值 或者 新加入的值大于这个最大值,那么更新最大值
        */
        int max_index = -1;
        int left = 0;
        int right = k - 1;
        while(right < len) {
            //遍历窗口寻找最大值的情况:第一个窗口还没有记录最大值、记录的最大值的索引超过了当前窗口
            if(max_index == -1 || max_index < left) {
                int max = nums[left];
                max_index = left;
                for(int i = left + 1;i <= right;i++) {
                    if(nums[i] > max) {
                        max = nums[i];
                        max_index = i;
                    }
                }
            }else if(nums[max_index] < nums[right]) { //更新最大值的情况:新加入的值比已经记录的最大值还大
                max_index = right;
            }
            arr[left] = nums[max_index];
            left++;
            right++;
        }
        return arr;
    }
}
上一篇 下一篇

猜你喜欢

热点阅读