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