您的位置:首页 > 编程语言

在线编程--生成窗口最大值数组

2016-04-04 17:11 260 查看
题目描述

给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。

例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5};

针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6个:

{[2,3,4],2,6,2,5,1},

{2,[3,4,2],6,2,5,1},

{2,3,[4,2,6],2,5,1},

{2,3,4,[2,6,2],5,1},

{2,3,4,2,[6,2,5],1},

{2,3,4,2,6,[2,5,1]}。

思路:利用双端队列实现更新。qmax是一个双端队列,,存放num数组的下标

遍历数组到num[i]时,qmax的规则为:

(1)如果qmax为空,直接把下标放入qmax队列,结束;

(2)如果qmax不为空,取出当前qmax队尾存放的下标,假设为j:

<1>如果num[j]>num[i],直接把下标放入qmax的队尾,结束;

<2>如果num[j]<=num[i],把j从qmax中弹出,继续qmax的存放规则。

import java.util.ArrayList;
import java.util.LinkedList;

public class Solution {
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
LinkedList<Integer> qmax=new LinkedList<Integer>();
ArrayList<Integer> res=new ArrayList<Integer>();//存放最后输出结果
if(num == null|| size<1 || num.length<size){
return res;
}
for(int i=0;i<num.length;i++){
while(!qmax.isEmpty()&&num[qmax.peekLast()]<num[i]){
qmax.pollLast(); //存大值下标,将大值存放入qmax靠前的位置
}
qmax.addLast(i);
if(qmax.peekFirst()==i-size){
qmax.pollFirst(); //超出队列所存放的size大小,去掉队列头
}
if(i>=size-1){
res.add(num[qmax.peekFirst()]);
//从滑动框大小处开始记录结果值(下标都是从0开始,故从size-1处开始存储)
}
}
return res;
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: