瞌睡 python实现

题目描述（链接）

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

输入描述:

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

输出描述:

• 小易这堂课听到的知识点的最大兴趣值。

输入：
6 3
1 3 5 2 5 4
1 1 0 1 0 0

输出：16

解题思路

n,k = 6,3
score = [1,3,5,2,5,4]
wake = [1,1,0,1,0,0]
一节课只叫醒小易一次，我们只需要关注wake里面0的情况，把所有的情况列出来，wake是0就叫醒，把当前连续k个值赋1，得到wake_新的清醒状态，根据wake_计算score的总分，取最大值。

def sleep(n,k,score,wake):
    lili = []
    for i in range(n):
        if min(wake)==1:
            ss = sum(score)
            break
        elif wake[i] == 0:
            wake_ = [0 for _ in range(n)]
            wake_[i:i+k] = [1 for _ in range(k)]
            wake_ = [wake[j] or wake_[j] for j in range(n)]
            s_ = sum([score[j] for j in range(n) if wake_[j]==1])
            lili.append(s_)
        if len(lili)!=0:
            ss = max(lili)
    return ss
n,k = 6,3
score = [1,3,5,2,5,4]
wake  = [1,1,0,1,0,0]
# wake  = [1,1,1,1,1,1]
sleep(n,k,score,wake)