关注
第三题(最多喝多少天咖啡): #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))
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 怎么给家人解释你的工作? #
11984次浏览 74人参与
# 牛客周边新品开箱 #
10677次浏览 91人参与
# 快手技术岗信息交流阵地 #
4173次浏览 36人参与
# 你的mentor是什么样的人? #
16764次浏览 112人参与
# 应届生被毁约被毁意向了怎么办 #
46462次浏览 280人参与
# 牛客树洞,我想对你说 #
10153次浏览 100人参与
# 机械人集合!你是什么工程师? #
20193次浏览 91人参与
# 求职中的尴尬瞬间 #
4346次浏览 44人参与
# 牛友的志愿填报指南 #
35208次浏览 185人参与
# 大疆工作体验 #
19009次浏览 85人参与
# 得物app工作体验 #
28698次浏览 67人参与
# 帮我看看,领导说这话什么意思? #
21973次浏览 98人参与
# 国企还是互联网,你怎么选? #
170570次浏览 1273人参与
# 三一集团提前批进度交流 #
38625次浏览 225人参与
# 如何KTV领导 #
72925次浏览 502人参与
# 校招泡的最久的公司是哪家? #
13535次浏览 84人参与
# 26届秋招公司红黑榜 #
28821次浏览 116人参与
# 硬件人的春招flag #
51895次浏览 435人参与
# 大学最后一个寒假,我想…… #
54076次浏览 598人参与
# 今年形式下双非本找得到工作吗 #
238028次浏览 1433人参与
# 实习期间如何提升留用概率? #
187298次浏览 1597人参与
