瞌睡 （网易笔试题）

题目描述

小易觉得高数课太无聊了，决定睡觉。不过他对课上的一些内容挺感兴趣，所以希望你在老师讲到有趣的部分的时候叫醒他一下。你知道了小易对一堂课每分钟知识点的感兴趣程度，并以分数量化，以及他在这堂课上每分钟是否会睡着，你可以叫醒他一次，这会使得他在接下来的k分钟内保持清醒。你需要选择一种方案最大化小易这堂课听到的知识点分值。

输入

第一行 n, k (1 <= n, k <= 10^5) ，表示这堂课持续多少分钟，以及叫醒小易一次使他能够保持清醒的时间。
第二行 n 个数，a1, a2, ... , an(1 <= ai <= 10^4) 表示小易对每分钟知识点的感兴趣评分。
第三行 n 个数，t1, t2, ... , tn 表示每分钟小易是否清醒, 1表示清醒。

输出

小易这堂课听到的知识点的最大兴趣值

输入样例 1 

6 3
1 3 5 2 5 4
1 1 0 1 0 0

输出样例 1

16                                                                                                                                                                                                    

题意：

 每分钟都会有两个量，一个是小易对这一分钟知识点的感兴趣评分，以及小易是否清醒，输入的k为叫醒小易后他能保持清醒的时间，只能叫醒小易一次，要求得出小易这堂课听到的知识点的最大兴趣值。

方法一：最先想到的就是一个很暴力的方法，先求出小易醒着的的兴趣值总和sum，然后从头开始遍历每一分钟，判断是否清醒，如果为不清醒，那么就遍历接下来k个值，把这k个值未醒着的分值加到前面的sum上，与记录的最大的兴趣值比较并不断更新最大值。但是这个绝对就会超时，确实超时了。

暴力超时代码：

import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
	   
		Scanner in=new Scanner(System.in);
		int n=in.nextInt();
		int k=in.nextInt();
		int[] value=new int[n];    //知识点的兴趣值
		int[] wake=new int[n];     //每分钟是否清醒
		for(int i=0;i<n;i++)
			value[i]=in.nextInt();
	    for(int i=0;i<n;i++)
	        wake[i]=in.nextInt();
	    int sum=0;
	    for(int i=0;i<n;i++)
	    {
	    	if(wake[i]==1)sum+=value[i];
	    }
	    int max=sum;
	    for(int i=0;i<n;i++)
	    {
	    	if(wake[i]==0)
	    	{
	    		int m=sum;
	    		for(int j=i;j<Math.min(i+k, n);j++)
	    		{
	    			if(wake[j]==0)m+=value[i];
	    		}
	    		if(m>max)max=m;
	    	}
	    	
	    }
	    System.out.println(max);
	}

}

方法二：可以定义3个数组，长度都为n。leftv数组是表示从左边开始到右清醒时兴趣值的累加和，rightv数组是表示从右边开始到左清醒时兴趣值的累加和，tolv表示从左到右所有兴趣值的累加和。

       然后当我们遍历到为未醒着的位置i时，那么这时的总的兴趣值为3部分之和：leftv[i-1] + rightv[i+k] + (tolv[i+k-1] - tolv[i-1])。最重要的就是考虑边界问题，第一部分的值为第i分钟左边的值，左边界的问题就是要考虑i-1不能小于0，如果小于0，那么左边的和就不是加leftv[i-1]而是加0，第二部分的值就是第i分钟右边得值，右边界的问题就是要考虑i+k要小于n，如果大于n，那么右边得和就不是加rightv[i+k]而是加0，最后一部分就是i与i+k之间的兴趣值的和，tolv[Math.min(i+k-1, n-1)]-(i<1?0:tolv[i-1])，也是需要考虑边界问题。

  可AC的代码：

import java.util.Scanner;

public class Main {

	public static void main(String[] args) {

		Scanner in=new Scanner(System.in);
		int n=in.nextInt();
		int k=in.nextInt();
		int[] value=new int[n];    //知识点的兴趣值
		int[] wake=new int[n];     //每分钟是否清醒
		for(int i=0;i<n;i++)
			value[i]=in.nextInt();
	    for(int i=0;i<n;i++)
	        wake[i]=in.nextInt();
	    int sum=0;
      
	    int[] leftv=new int[n];            //从左边开始统计此分钟之前清醒时所有的兴趣值
	    for(int i=0;i<n;i++){
	    	if(wake[i]==1)sum+=value[i];
	    	leftv[i]=sum;
	    } 
       
	    int[] rightv=new int[n];         //从右边开始统计此分钟之前清醒时所有的兴趣值
	    sum=0;
	    for(int i=n-1;i>=0;i--)
	    {
	    	if(wake[i]==1)sum+=value[i];
	    	rightv[i]=sum;
	    }
      
	    int[] tolv=new int[n];         //记录此分钟之前的清醒和未清醒所有的兴趣值
	    sum=0;
	    for(int i=0;i<n;i++)
	    {
	       sum+=value[i];
	       tolv[i]=sum;
	    }
      
	    if(n==0)                 //如果n为0，结果一定为0
	    {
	    	System.out.println(0); 
	    }
	    else if(k==0)                 //如果K为0，那么结果就是所有清醒时的兴趣值
	    {
	    	System.out.println(leftv[n-1]);
	    }
	    else                         //n不为0，K也不为0
	    {
	    	int res=0;
		    for(int i=0;i<n;i++)
		    {
		    	if(wake[i]==0)
		    	{
		    		int m=0;
		    		if(i-1<0)m+=0;      //考虑左边界的问题，
		    		else m+=leftv[i-1];
		    		if(i+k>=n)m+=0;     //考虑右边界的问题
		    		else m+=rightv[i+k];
		    		m+=tolv[Math.min(i+k-1, n-1)]-(i<1?0:tolv[i-1]);
		    		if(m>res)res=m;
		    	}
		    }
	        System.out.println(res);
	    }
	}

}