小易觉得高数课太无聊了,决定睡觉。不过他对课上的一些内容挺感兴趣,所以希望你在老师讲到有趣的部分的时候叫醒他一下。你知道了小易对一堂课每分钟知识点的感兴趣程度,并以分数量化,以及他在这堂课上每分钟是否会睡着,你可以叫醒他一次,这会使得他在接下来的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;
}