窗口最大值数组

生成窗口最大值数组

有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边，窗口每次向右边滑一个位置。

例如，数组为[4,3,5,4,3,3,6,7]，窗口大小为3时：

 

[4 3 5] 4 3 3 6 7 窗口中最大值为5

4 [3 5 4] 3 3 6 7 窗口中最大值为5

4 3 [5 4 3] 3 6 7 窗口中最大值为5

4 3 5 [4 3 3] 6 7 窗口中最大值为4

4 3 5 4 [3 3 6] 7 窗口中最大值为6

4 3 5 4 3 [3 6 7] 窗口中最大值为7

结果应该返回[5,5,5,4,6,7]。

 

简单代码如下：

/**
 * 简单方法
 * 狗剩的美丽家园
 */
#include <iostream>
#include <vector>
using namespace std;

class Solution {
public:
    int getMax(const int A[], int size) {
        int max_num = A[0];
        for (int i = 0; i < size; i++) {
            if (A[i] > max_num) {
                max_num = A[i];
            }
        }
        return max_num;
    }

    vector<int> maxInWindows(const int A[], int n, int size) {
        vector<int> result;
        for (int i = 0; i < n - size + 1; i++) {
            int num = getMax(A + i, size);
            result.push_back(num);
        }
        return result;
    }
};
int main() {
    Solution s;
    const int A[] = {2,3,4,2,6,2,5,1};
    int size = 3;
    int n = sizeof(A) / sizeof(A[0]);
    vector<int> res;
    res = s.maxInWindows(A, n, size);
    for (int i = 0; i < res.size(); i++) {
        cout << res[i] << ",";
    }
}

时间复杂度为 O(N)的解法如下：

/**
 * 双端队列
 * 狗剩的美丽家园
**/

#include <iostream>
#include <vector>
#include <deque>
using namespace std;

vector<int> maxInWindows(int A[], int n, int size) {
    deque<int> Q;
    vector<int> result;
    for (int i = 0; i < n; i++) {
        //当前元素大于等于队尾，队尾出队
        while (!Q.empty() && A[Q.back()] <= A[i]) {
            Q.pop_back();
        }
        Q.push_back(i);

        //窗口范围大于size，队头出队列
        if (Q.front() == i - size) {
            Q.pop_front();
        }
        if (i >= size - 1) {
            result.push_back(A[Q.front()]);
        }
    }
    return result;
}

int main() {
    int A[] = {2,3,4,2,6,2,5,1};
    int size = 3;
    int n = sizeof(A) / sizeof(A[0]);
    vector<int> res;
    res = maxInWindows(A, n, size);
    for (int i = 0; i < res.size(); i++) {
        cout << res[i] << ",";
    }
}