关注
第三题(最多喝多少天咖啡): #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))
查看原帖
点赞 评论
相关推荐
11-19 18:40
安徽工业大学 网络安全 点赞 评论 收藏
分享
牛客热帖
正在热议
# 25届秋招总结 #
296614次浏览 2610人参与
# 美团求职进展汇总 #
1326012次浏览 12438人参与
# 北方华创开奖 #
26165次浏览 281人参与
# 地方国企笔面经互助 #
3695次浏览 9人参与
# 如果不工作真的会快乐吗 #
58674次浏览 508人参与
# 选完offer后,你后悔学本专业吗 #
19359次浏览 140人参与
# 百度开奖 #
159416次浏览 955人参与
# 国央企薪资爆料 #
7526次浏览 55人参与
# 学历or实习经历,哪个更重要 #
50549次浏览 398人参与
# 海康威视求职进展汇总 #
398491次浏览 3405人参与
# 正在实习的你,几点下班 #
51377次浏览 382人参与
# 米哈游求职进展汇总 #
175647次浏览 1456人参与
# 投递实习岗位前的准备 #
1178613次浏览 18388人参与
# 面试体验感最好的是哪家? #
84950次浏览 845人参与
# 如何一边实习一边秋招 #
991275次浏览 12634人参与
# 得物求职进展汇总 #
66071次浏览 680人参与
# 实习生应该准时下班吗 #
167305次浏览 1159人参与
# 求职遇到的搞笑事件 #
70539次浏览 575人参与
# 网申一定要掌握的小技巧 #
5295次浏览 52人参与
# 0offer是寒冬太冷还是我太菜 #
897387次浏览 8006人参与
# 腾讯求职进展汇总 #
195519次浏览 1639人参与
# 提前批简历挂麻了怎么办 #
146249次浏览 1944人参与