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

输出

16

说明

 

只能叫醒一次，直接处理，时间复杂度O(Nk), 可以通过90%

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

const int N = 100010;

int nums[N];
int t[N];


int helper(int i, int n, int k)
{
	int j = 0;
	int res = 0;
	while (i + j < n && j < k) {
		if (t[i + j] == 0)
			res += nums[i + j];
		j++;
	}
	return res;
}

int main()
{
	int n, k;
	cin >> n >> k;
	for (int i = 0; i < n; i++) cin >> nums[i];
	for (int i = 0; i < n; i++) cin >> t[i];
	int sum = 0;
	for (int i = 0; i < n; i++) {
		if (t[i] == 0) {
			int temp = helper(i, n, k);
			sum = max(sum, temp);
		}
	}
	int res = 0;
	for (int i = 0; i < n; i++) {
		if (t[i]) {
			res += nums[i];
		}
	}
	cout << res + sum << endl;
}

 

使用滑动窗口的办法，计算连续k个0对应的最大值之和，时间复杂度优化到O(N)

通过。 注意，这里窗口大小不一定就是k，所以循环的时候可以用补0的技巧处理。

总的来说，这道题目偏向模拟。

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

const int N = 100010;

int nums[N];
int t[N];


int helper(int i, int n, int k)
{
	int j = 0;
	int res = 0;
	while (i + j < n && j < k) {
		if (t[i + j] == 0)
			res += nums[i + j];
		j++;
	}
	return res;
}

int main()
{
	int n, k;
	cin >> n >> k;
	for (int i = 0; i < n; i++) cin >> nums[i];
	for (int i = 0; i < n; i++) cin >> t[i];
	int sum = 0;
	for (int i = 0; i < n; i++) {
		if (t[i] == 0) {
			int temp = helper(i, n, k);
			sum = max(sum, temp);
		}
	}
	int res = 0;
	for (int i = 0; i < n; i++) {
		if (t[i]) {
			res += nums[i];
		}
	}
	cout << res + sum << endl;
}