算法--滑动窗口

转载请标明出处http://www.cnblogs.com/haozhengfei/p/a14049ec0869a8125a69f3af37471c77.html 

---

滑动窗口练习题

 

第8节 滑动窗口练习题

 

有一个整型数组 arr 和一个大小为 w 的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。 返回一个长度为n-w+1的数组res，res[i]表示每一种窗口状态下的最大值。 以数组为[4,3,5,4,3,3,6,7]，w=3为例。因为第一个窗口[4,3,5]的最大值为5，第二个窗口[3,5,4]的最大值为5，第三个窗口[5,4,3]的最大值为5。第四个窗口[4,3,3]的最大值为4。第五个窗口[3,3,6]的最大值为6。第六个窗口[3,6,7]的最大值为7。所以最终返回[5,5,5,4,6,7]。

给定整形数组arr及它的大小n，同时给定w，请返回res数组。保证w小于等于n，同时保证数组大小小于等于500。

测试样例：

[4,3,5,4,3,3,6,7],8,3

返回：[5,5,5,4,6,7]

Java (javac 1.7)

代码自动补全

 

 

 

 

 

1

import java.util.*;

2

3

public class SlideWindow {

4

    public int[] slide(int[] arr, int n, int w) {

5

        //result数组中保存每个窗口状态下的最大值

6

        int[] result = new int[n-w+1];

7

8

        //记录双端队列队头的下标 ，队尾下标

9

        int[] qmax = new int[n];

10

        int front = 0, back = 0;

11

12

        //j 标记是否达到窗口大小,同时记录result中下一个应该放入的元素的下标

13

        int j = 0;

14

15

        for(int i=0; i<n; i++){

16

17

            while(front < back && arr[qmax[back-1]] < arr[i])//back为当前要往qmax中放入的值

18

                back--;

19

            qmax[back++] = i;

20

21

            if(j+w-1 == i){

22

                //达到窗口长度

23

                result[j] = arr[qmax[front]];

24

                j++;

25

            }

26

            if(qmax[front]+w-1 == i){

27

                //队头过期

28

                front++;

29

            }

30

        } 

31

        return result;

32

    }

33

}

 

 

您的代码已保存
答案正确:恭喜！您提交的程序通过了所有的测试用例

运行

 

 

代码示例

package com.hzf.stack$queue;

/**
 * 有一个整型数组 arr 和一个大小为 w 的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。
 * 返回一个长度为n-w+1的数组res，res[i]表示每一种窗口状态下的最大值。
 * 以数组为[4,3,5,4,3,3,6,7]，w=3为例。因为第一个窗口[4
 * ,3,5]的最大值为5，第二个窗口[3,5,4]的最大值为5，第三个窗口[5,4,
 * 3]的最大值为5。第四个窗口[4,3,3]的最大值为4。第五个窗口[3,3
 * ,6]的最大值为6。第六个窗口[3,6,7]的最大值为7。所以最终返回[5,5,5,4,6,7]。
 * 
 * 给定整形数组arr及它的大小n，同时给定w，请返回res数组。保证w小于等于n，同时保证数组大小小于等于500。
 * 
 * 测试样例： [4,3,5,4,3,3,6,7],8,3 返回：[5,5,5,4,6,7]
 * 
 * @author hzf
 *
 */
public class SlideWindow {
    public int[] slide(int[] arr, int n, int w) {
        //result数组中保存每个窗口状态下的最大值
        int[] result = new int[n-w+1];
        
        //记录双端队列队头的下标 ，队尾下标
        int[] qmax = new int[n];
        int front = 0, back = 0;
        
        //j 标记是否达到窗口大小,同时记录result中下一个应该放入的元素的下标
        int j = 0;
        
        for(int i=0; i<n; i++){
            
            while(front < back && arr[qmax[back-1]] < arr[i])//back为当前要往qmax中放入的值
                back--;
            qmax[back++] = i;
             
            if(j+w-1 == i){
                //达到窗口长度
                result[j] = arr[qmax[front]];
                j++;
            }
            /**
             * 保证qmax中front所代表的索引不会过期，
             * 这个索引的边界就是：qmax[front]+w-1 == i，满足这个条件front++，否则下一窗口就会过期
             * 比如
             *          front  back
             *   qmax     6     8
             *   arr      11    5
             *   当窗口大小w==3时,假设此时result[j]取arr[qmax[front]],front没有更新，
             *   如果下一次的arr[9] = 4,那么有如下   
             *           front      back
             *   qmax     6     8     9
             *   arr      11    5     4
             *   此时如果在上一步中没有更新front,那么此时窗口[7,8,9]的最大值取的是arr[qmax[6]]，显然6已经过期
             *   
             *   所以综上所述，一旦qmax[front]所代表的索引达到该索引所能控制的边界，front++，以防下一次该front已经过期
             */
            if(qmax[front]+w-1 == i){
                //队头过期
                front++;
            }    
        } 
        return result;
    }
}