网易面试题 瞌睡

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

输入描述:

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

输出描述:

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

输入例子1:

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

输出例子1:

16

解法一:

#include <iostream>
#include <vector>
using namespace  std;

int main(){
	int n,k;
	cin>>n>>k;
	vector<int> score(n,0);
	vector<int> sleep(n,0);
	
	for(int i=0;i<n;i++)
		cin>>score[i];
	
	int alreScore = 0;
	for(int i=0;i<n;i++){
		cin>>sleep[i];
		if(sleep[i] == 1){
			alreScore += score[i];//先计算原来能得的最高分 
			score[i] = 0;
		}	
	} 
		
	int maxScore = 0;
	for(int i = 0;i < k;i++){
		maxScore += score[i];//计算从第一分钟叫醒的最高分	 
	}
	
	int getScore = maxScore;
	for(int i = 1;i <= n-k;i++){//注意newscore的赋值只能在外面一次!!!向右循环计算三个数的结果 
		getScore -= score[i-1];
		getScore += score[i+k-1];
		if(getScore > maxScore)
			maxScore = getScore;
	}	

	cout<<maxScore + alreScore<<endl;
	
	return 0;
}

解法二:

#include <iostream>
#include <vector>
using namespace  std;

int main(){
	int n,k;
	cin>>n>>k;
	vector<int> score(n,0);
	vector<int> sleep(n,0);
	
	for(int i=0;i<n;i++)
		cin>>score[i];
	
	int alreScore = 0;
	for(int i=0;i<n;i++){
		cin>>sleep[i];
		if(sleep[i] == 1)
			alreScore += score[i];//先计算原来能得的最高分 
	} 
		
	int maxScore = alreScore;
	for(int i = 0;i < k;i++){
		if(sleep[i] == 0){
			maxScore += score[i];//计算从第一分钟叫醒的最高分
		}	 
	}
	
	int newscore = maxScore;//注意newscore的赋值只能在外面一次!!!不然会出现逻辑计算错误 
	for(int i = 1;i <= n-k;i++){
		if(sleep[i-1] == 0)
			newscore -= score[i-1];
		if(sleep[i+k-1] == 0)
			newscore += score[i+k-1];
		if(newscore > maxScore)
			maxScore = newscore;
	}	

	cout<<maxScore<<endl;
	
	return 0;
}