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

来源:互联网 发布:如何当淘宝模特 编辑:程序博客网 时间:2024/06/03 15:01

题目:给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]);        }    }}
原创粉丝点击