牛客网-瞌睡

Python 代码

def f(n,k,a,t):

# k>=n,整节课都保持清醒
if k>=n:
return sum(a)

"""
#将评分分为两部分：
#（1）一部分是没有叫醒服务，得到的评分，cur
#（2）一部分是叫醒一次，额外得到的评分，other
"""
cur=0
other=[0]
for i in range(n):
cur+=a[i]*t[i]
other.append(other[i]+a[i]*(1-t[i]))

M=0
for i in range(n):
if i+k-1<n:
M=max(M,other[i+k]-other[i])
else:
#越界
M=max(M,other[-1]-other[i])
return cur+M

n,k=list(map(int,input().split()))
a=list(map(int,input().split()))
t=list(map(int,input().split()))
print(f(n,k,a,t))

• 点赞 1
• 收藏
• 分享
• 文章举报