校招真题练习025 瞌睡(网易)

瞌睡

题目描述
小易觉得高数课太无聊了，决定睡觉。不过他对课上的一些内容挺感兴趣，所以希望你在老师讲到有趣的部分的时候叫醒他一下。你知道了小易对一堂课每分钟知识点的感兴趣程度，并以分数量化，以及他在这堂课上每分钟是否会睡着，你可以叫醒他一次，这会使得他在接下来的k分钟内保持清醒。你需要选择一种方案最大化小易这堂课听到的知识点分值。

输入描述:
第一行 n, k (1 <= n, k <= 105) ，表示这堂课持续多少分钟，以及叫醒小易一次使他能够保持清醒的时间。
第二行 n 个数，a1, a2, ... , an(1 <= ai <= 104) 表示小易对每分钟知识点的感兴趣评分。
第三行 n 个数，t1, t2, ... , tn 表示每分钟小易是否清醒, 1表示清醒。

输出描述:
小易这堂课听到的知识点的最大兴趣值。

 1 n,k = [int(x) for x in input().strip().split()]
 2 scores = [int(x) for x in input().strip().split()]
 3 wakes = [int(x) for x in input().strip().split()]
 4 
 5 if n <= k:
 6     print(sum(scores))
 7 else:
 8     maxscore = 0
 9     sums = 0
10     for i in range(n):
11         if wakes[i] == 1:
12             sums += scores[i]
13             scores[i] = 0
14     i = 0
15     j = 0
16     while j < k:
17         if wakes[j] == 0:
18             sums += scores[j]
19         j += 1
20     maxscore = max(maxscore,sums)
21     for i in range(n-k):
22         sums -= scores[i]
23         sums += scores[j]
24         j += 1
25         maxscore = max(maxscore,sums)
26     print(maxscore)

算法思路：滑动窗口。

在滑动窗口内计算全部值的和。窗口滑动时，将左侧的值减去，将右侧的值加上。每次移动窗口计算当前窗口内的得分，保留最大值。