链接:https://www.nowcoder.com/questionTerminal/93f2c11daeaf45959bb47e7894047085?toCommentId=1620177
来源:牛客网
输入:
第一行 n, k (1 <= n, k <= 105) ,表示这堂课持续多少分钟,以及叫醒小易一次使他能够保持清醒的时间。
第二行 n 个数,a1, a2, … , an(1 <= ai <= 104) 表示小易对每分钟知识点的感兴趣评分。
第三行 n 个数,t1, t2, … , tn 表示每分钟小易是否清醒, 1表示清醒。
输出:
小易这堂课听到的知识点的最大兴趣值。
滑动窗口法:
先有一个初值,再依次滑动
#include<iostream>
#include<vector>
using namespace std;
int main()
{
int n,k;
cin>>n>>k;
vector<int>instrest(n);
vector<int>flag(n);
for (int i=0;i<n;i++)cin>>instrest[i];
for (int i=0;i<n;i++)cin>>flag[i];
int score = 0;
int max_score=0;
int ans = 0;
for (int i=0;i<k;i++)
{
if (flag[i]==0)score+=instrest[i];
else ans += instrest[i];
}
max_score = score;
for (int i=k;i<instrest.size();i++)
{
if (flag[i]==0)score += instrest[i];
else ans += instrest[i];
if (flag[i-k]==0)score -= instrest[i-k];
max_score = max(max_score,score);
}
ans += max_score;
cout<<ans;
return 0;
}