(滑动窗口)瞌睡

链接: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;
}

猜你喜欢

转载自blog.csdn.net/xnmc2014/article/details/88566345
今日推荐