关注
第三题(最多喝多少天咖啡): #coding=utf-8
import sys
def func(k, m, days):
if k == 0:
return 30
if m == 0:
return len(range(1, 31, k + 1))
# 初始化:长度为30的列表,将固定喝咖啡的日子置为1,其他为0
dp = [1 if i + 1 in days else 0 for i in range(30)]
for i in range(30):
if dp[i] == 1:
continue
# 如果第i天前面k天没喝咖啡,同时后面k天也没喝,那第i天就可以喝,置为1
if sum(dp[max(0, i - k) : i]) == 0 and sum(dp[i+1 : i+1+k]) == 0:
dp[i] = 1
return sum(dp)
if __name__ == "__main__":
T = int(sys.stdin.readline().strip())
for _ in range(T):
k, m = list(map(int, sys.stdin.readline().strip().split()))
days = list(map(int, sys.stdin.readline().strip().split()))
print(func(k, m, days))
查看原帖
点赞 评论
相关推荐
牛客热帖
正在热议
# 牛友的五一计划 #
28261次浏览 487人参与
# 市场营销面经 #
1878次浏览 82人参与
# 牛客帮帮团来啦!有问必答 #
424400次浏览 8065人参与
# 许愿池 #
67471次浏览 1489人参与
# 晒一晒我的offer #
2874007次浏览 50275人参与
# 2022届毕业生现状 #
287631次浏览 4122人参与
# 你的秋招进展怎么样了 #
451045次浏览 12996人参与
# 如何看待offer收割机的行为 #
199442次浏览 3041人参与
# 互联网公司评价 #
64281次浏览 912人参与
# 实习好累,可以辞职全力准备秋招吗 #
2829次浏览 60人参与
# 硬件人的春招flag #
14807次浏览 203人参与
# 非技术岗薪资爆料 #
11221次浏览 211人参与
# 找工作,你会甘心进小厂还是猛冲大厂 #
26055次浏览 262人参与
# 23届的你们都什么时候入职? #
77483次浏览 692人参与
# 产品实习,你更倾向大公司or小公司 #
31752次浏览 499人参与
# 秋招开了,你想投哪些公司呢 #
102824次浏览 3135人参与
# 浅聊一下我实习的辛苦费 #
71884次浏览 656人参与
# 提前批真的不会影响正式批吗 #
18394次浏览 231人参与
# 双非本科求职如何逆袭 #
176186次浏览 2643人参与
# 在国企工作的人,躺平了吗? #
74886次浏览 919人参与