滑动窗口 AcWing(JAVA)

 给定一个大小为 n≤10^6 的数组。

有一个大小为 k 的滑动窗口,它从数组的最左边移动到最右边。

你只能在窗口中看到 k 个数字。

每次滑动窗口向右移动一个位置。

以下是一个例子:

该数组为 [1 3 -1 -3 5 3 6 7],k 为 33。

窗口位置最小值最大值
[1 3 -1] -3 5 3 6 7-13
1 [3 -1 -3] 5 3 6 7-33
1 3 [-1 -3 5] 3 6 7-35
1 3 -1 [-3 5 3] 6 7-35
1 3 -1 -3 [5 3 6] 736
1 3 -1 -3 5 [3 6 7]37

 你的任务是确定滑动窗口位于每个位置时,窗口中的最大值和最小值。

输入格式:

 输入包含两行。

第一行包含两个整数 n和 k,分别代表数组长度和滑动窗口的长度。

第二行有 n 个整数,代表数组的具体数值。

同行数据之间用空格隔开。

输出格式:

输出包含两个。

第一行输出,从左至右,每个位置滑动窗口中的最小值。

第二行输出,从左至右,每个位置滑动窗口中的最大值。

解题思路: 如果朴素的做法,将每次得到的窗口进行遍历找最小值。时间复杂度大概是nk,不出意外是会超时的。如何对其进行优化呢?

首先以数组q[N],作为本题队列,a[N]存入输入的元素。我们知道下标和元素具有一 一对应的关系,所以q[i]不是存入元素所对应的值,而是元素所对应的下标。

 在这里设hh = 0为头指针,tt = - 1为尾指。

首先应该明确窗口的特点:即内部元素的数量在k的时候即可输出最内部最小值和最大值。此后每移动一下就再次输出一次。这恰好哦可以用循环变量i来表示,如下窗口:

[a,b,c,d.....]

0                i             窗口内部元素的数量可用 (i - 0 + 1)来表示,当窗口内数量够k个的时候即可输出,即i - 0 + 1 >= k即 i >= k - 1。

有趣的是当 0~i + 2 中c为最小值的时候,当窗口向右边移动的时候:

[b,c,d,e.....]

1               i + 1         min = c

[c,d,e,f.....]

2              i + 2          min = c

不难看出窗口在移动的时候,最小值仍是c,如果能用队列存入最小值的下标(即q[tt] = 2),那就避免重新遍历窗口的麻烦。那问题来了设立队列只是为了存入一个最小值的吗?这显然不是,如下窗口继续向右移动:

[d,e,f,g.....]

3             i + 3       min =  ?    由此可见如果队列里值只存入c的下标,当窗口不包含c的时候队列内的 q[tt] = 2 也就没用了。

那么需要队列,理想的情况是让队列内存入 最小元素下标,第二小元素下标,第三小元素下标.......

当窗口不再包含最小元素的时候那么就可以在队列内删去了,那么第二小元素就理应成为新的最小元素了,以此类推,队列头部元素一直是符合条件的最小元素下标。

至于如何输出最大值,和上述反着来即可。

理论成立代码如下(未加速版):

import java.util.*;
 
public class Main {
    public static int N = 1000010;
    public static void main(String[] args) {
      Scanner sc  = new Scanner(System.in);
      int a[] = new int [N];//数组
      int q[] = new int [N];//队列
      int n = sc.nextInt();
      int k = sc.nextInt();
      for(int i = 0; i < n; i ++) a[i] = sc.nextInt();
      int hh = 0, tt = -1;//队头,队尾,q[hh]存入最小元素下标
      for(int i = 0; i < n; i ++) {//找最小值
    	  if(hh <= tt && q[hh] < i - k + 1) hh ++;//在滑动窗口的外面,移动左窗口
    	  while(hh <= tt && a[q[tt]] >=a[i]) tt--;//队尾不比添加进来的数小,那就没用,删掉。
    	  q[++ tt] = i;//添加元素。
    	  if(i >= k - 1) {
    		  if(i == n-1)
    		  System.out.println(a[q[hh]]);
    		  else
    			  System.out.print(a[q[hh]]+" ");  
    	  }	    
      }
//输出最大值
      hh = 0; tt = -1;
      for(int i = 0; i < n; i ++) {
    	   if(hh <= tt && q[hh] < i - k + 1 ) hh ++;
    	   while(hh <= tt && a[q[tt]] <= a[i]) tt --;
    	   q[++ tt] = i;
    	   if(i >= k - 1) {
    		   if(i == n - 1)
    			   System.out.print(a[q[hh]]);
    		   else
    			   System.out.print(a[q[hh]]+" ");
    	   }
      }
	  }
}

由于最后一个数据过于庞大,时间超时了,想通过,得用如下的加速版:

import java.io.*;
import java.util.*;;
public class Main {
    public static int N = 1000010;
	public static void main(String[] args) throws IOException {
		// TODO Auto-generated method stub
		Scanner sc = new Scanner(new BufferedInputStream(System.in));
        BufferedWriter sout = new BufferedWriter(new OutputStreamWriter(System.out));
        int a[] = new int [N];//数组
        int q[] = new int [N];//队列
        int n = sc.nextInt();
        int k = sc.nextInt();
        for(int i = 0; i < n; i ++) a[i] = sc.nextInt();
        int hh = 0, tt = -1;//队头,队尾,q[hh]存入最小元素下标
        for(int i = 0; i < n; i ++) {//找最小值
      	  if(hh <= tt && q[hh] < i - k + 1) hh ++;//在滑动窗口的外面,移动左窗口
      	  while(hh <= tt && a[q[tt]] >=a[i]) tt--;//队尾不比添加进来的数小,那就没用,删掉。
      	  q[++ tt] = i;//添加元素。
      	  if(i >= k - 1) {
      			sout.write(a[q[hh]]+" ");  
      	  }	    
        }
        sout.write("\n");
        //输出最大值
        hh = 0; tt = -1;
        for(int i = 0; i < n; i ++) {
      	   if(hh <= tt && q[hh] < i - k + 1 ) hh ++;
      	   while(hh <= tt && a[q[tt]] <= a[i]) tt --;
      	   q[++ tt] = i;
      	   if(i >= k - 1) {
      			 sout.write(a[q[hh]]+" ");
      	   }
        }
        sout.close();
        sc.close();
	}

}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值