题目描述

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

解题思路

采用滑动窗口的思想，所获得的最大兴趣值为：未叫醒之前所得的兴趣值+滑动窗口中兴趣值。

完整代码

n, k = map(int, input().split())
A = list(map(int, input().split()))
T = list(map(int, input().split()))

l = [0 for i in range(n + 1)]
for j in range(n):
    l[j + 1] = l[j] + A[j] * T[j]
res = 0

if n > k:
    for i in range(n):
        if T[i] == 0:
            if i + k < n:
                res =  max(res, l[n] + sum(A[i: i + k]) - l[i + k] + l[i + 1])
            else:
                res =  max(res, l[n] + sum(A[i:]) - l[n] + l[i + 1])
else:
    res = sum(A)
print(res)