瞌睡--牛客网--面试题

题目描述

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

输出

16

 

参考代码：

#include <cstdio>
#include <iostream>
using namespace std;
const int MAXN = 1e5 + 5;
 
int maxInterest, n, k;
int interest[MAXN], awake[MAXN];
 
void read() {
	scanf("%d%d", &n, &k);
	for (int i = 1; i <= n; i++) scanf("%d", &interest[i]);
	for (int i = 1; i <= n; i++) scanf("%d", &awake[i]);
}
 
void solve() {
	//最大兴趣值 = 清醒时的兴趣值 + k分钟叫醒的额外最大兴趣值
	for (int i = 1; i <= n; i++) if (awake[i]) maxInterest += interest[i];
 
	if (n < k) k = n; //
	int curInterest = 0;
	/*前缀和*/
	//先计算前k分钟叫醒的额外兴趣值
	for (int i = 1; i <= k; i++) if (0 == awake[i]) curInterest += interest[i];
	//向后移动时计算k分钟叫醒的最大额外兴趣值
	int _maxInterest = curInterest;
	for (int i = k + 1; i <= n; i++) {
		if (0 == awake[i]) curInterest += interest[i];
		if (0 == awake[i - k]) curInterest -= interest[i - k];
		if (curInterest > _maxInterest) _maxInterest = curInterest;
		
	}
	maxInterest += _maxInterest;
	printf("%d\n", maxInterest);
}
 
int main() {
	read();
	solve();
	return 0;
}