2019 网易校招编程题解答

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

 

这道题目因为只能叫醒一次，所以可以采用穷举的方法，醒来后非0之和最大。时间，复杂度是O(N^2)，这样可以通过90%的数据

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

const int N = 100100;
int main()
{
    int arr[N], t[N];
	vector<int> mp;
	int n, k, ans = 0;
	scanf("%d %d", &n, &k);
	for (int i = 0; i < n; i++) 
		scanf("%d", &arr[i]);
	
	
	for (int i = 0; i < n; i++) {
		scanf("%d", &t[i]);
		if (t[i] == 0)
			mp.push_back(i);
		else
			ans += arr[i];
	}
	
	int sum=0;
	int res = 0;
	int size = mp.size();
	for (int i = 0; i < size; i++) {
        sum = 0;
		for (int j = i; j < size && mp[j]-mp[i]<k; j++) {
            sum += arr[mp[j]];
		}
		//cout << sum << endl;
		res = max(res, sum);
        
	}
	
	printf("%d\n", res+ans);
}