滑动窗口的最大值(C++/Java)

题目:给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5};
思路:用一个双端队列,我们并不把每一个值都存入双端队列,而是 把可能成为最大值的数值的下标存入队列的中,对头每次保存的是当前的最大值。首先在size返回内初始化队列,如果当前数组的值大于等于队列的尾端的值,则把队列尾端的值剔除出去,加入当前的值得下标。
接下来就是每次数组的值先和队列尾端的值比较,如果大于等于尾端的值则把尾端的值剔除出去,接下来和队列的头比较,如果队列的头中存的下标值不在窗口内(i-头下标>=size)则把头下标剔除出去,加入当前值的下标。
c++实现

#include <iostream>
#include <queue>
#include <string>
using namespace std;
//num原始数组, width窗口大小
vector<int> maxInWindows(const vector<int>& num, unsigned  width) {
    vector<int> result;
    if(num.size() >= width && width >= 1) {
        deque<int> index;
        for(unsigned i = 0; i < width; i++) {
            while(!index.empty() && num[i] >= num[index.back()]) {
                index.pop_back();
            }
            index.push_back(i);
        }
        for(unsigned i = width; i < num.size(); i++) {
            result.push_back(num[index.front()]);
            while(!index.empty() && num[i] >= num[index.back()]) {
                index.pop_back();
            }
            index.push_back(i);
            if((int)(i - width) >= index.front()) {
                index.pop_front();
            }
            index.push_back(i);
        }
        result.push_back(num[index.front()]);
    }
    return result;
}

int main()
{
    int arr[] = {4, 3, 5, 4, 3, 3, 6, 7};
    vector<int> num(arr, arr+8);
    vector<int> result = maxInWindows(num, 3);
    for(unsigned i = 0; i < result.size(); i++) {
        cout << result[i] << " ";
    }
    cout << endl;
    return 0;
}

Java实现

import java.util.LinkedList;
public class GetMaxWindow {

    /**
     * 
     * @param arr 原始数组
     * @param w 窗口大小
     * @return 求的窗口最大值数组
     */
    public static int[] getMaxWindow(int[]arr, int w) {
        if(arr == null || w < 1 || arr.length < w) {
            return null;
        }
        LinkedList<Integer> qmax = new LinkedList<>();
        int []res = new int[arr.length - w + 1];
        int index = 0;
        for(int i = 0; i < arr.length; i++) {
            while(!qmax.isEmpty() && arr[qmax.peekLast()] <= arr[i]) {
                qmax.pollLast();
            }
            qmax.addLast(i);
            if(qmax.peekFirst() == i - w) {
                qmax.pollFirst();
            }
            if(i >= w - 1) {
                res[index++] = arr[qmax.peekFirst()];
            }
        }
        return res;
    }

    public static void main(String[] args) {
        int arr[] = {4, 3, 5, 4, 3, 3, 6, 7};
        int res[] = getMaxWindow(arr, 3);
        for(int i = 0; i < res.length; i++) {
            System.out.println(res[i]);
        }
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值