题目:给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。
思路:维护一个队列,保证里面元素和最大 队列里存的是下标!!!
vector<int> maxInWindows(const vector<int>& num, unsigned int size) { vector<int> vec; if(num.size()<size) return vec; deque<int> Q; for(int i=0;i<num.size();i++){ if(Q.empty()){ Q.push_back(i); }else{ while(!Q.empty()&&num[Q.back()]<num[i]) Q.pop_back(); Q.push_back(i); } if(Q.back()-Q.front()>=size) Q.pop_front(); if(size==1||i>size-2)//size为无符号数,size=1时,size-2会是一个很大的数 vec.push_back(num[Q.front()]); } return vec; }
java版 队列里存的是下标!!!
public class No65 { public static void main(String[] args) { ArrayList<Integer> res= maxInWindows(new int[]{2,3,4,2,6,2,5,1},3); System.out.println(res); } public static ArrayList<Integer> maxInWindows(int [] num, int size) { ArrayList<Integer> res = new ArrayList<>(); if (num == null || num.length == 0 || size == 0 || size > num.length) { return res; } LinkedList<Integer> deque = new LinkedList<>(); for (int i = 0; i < num.length; i++) { if (!deque.isEmpty()) { // 如果队列头元素不在滑动窗口中了,就删除头元素 if (i >= deque.peek() + size) { deque.pop(); } // 如果当前数字大于队列尾,则删除队列尾,直到当前数字小于等于队列尾,或者队列空 while (!deque.isEmpty() && num[i] >= num[deque.getLast()]) { deque.removeLast(); } } deque.offer(i); // 入队列 // 滑动窗口经过一个滑动窗口的大小,就获取当前的最大值,也就是队列的头元素 if (i + 1 >= size) { res.add(num[deque.peek()]); } } return res; } }