关注
第三题(最多喝多少天咖啡): #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))
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
点赞 评论 收藏
分享
05-21 15:47
门头沟学院 Java 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 26届校招投递进展 #
30332次浏览 233人参与
# 小米提前批笔试难吗 #
33998次浏览 357人参与
# 现代汽车前瞻技术研发急速编程挑战赛 #
11801次浏览 127人参与
# 为了找工作你花了哪些钱? #
27736次浏览 262人参与
# 央国企投递记录 #
88035次浏览 1359人参与
# 神州信息工作体验 #
11576次浏览 56人参与
# 你觉得专业和学校哪个对薪资影响最大 #
61331次浏览 490人参与
# 设计人的面试记录 #
123380次浏览 1341人参与
# 来聊聊你目前的求职进展 #
634227次浏览 6745人参与
# 外包能不能当跳板? #
34379次浏览 220人参与
# 你今年的保底offer是哪家 #
118348次浏览 537人参与
# 烟草笔面经互助 #
16893次浏览 180人参与
# 大疆的机械笔试比去年难吗 #
72893次浏览 618人参与
# 打工人的精神状态 #
49534次浏览 858人参与
# 牛友们,签完三方你在忙什么? #
98220次浏览 852人参与
# 听到哪句话就代表面试稳了or挂了? #
170762次浏览 1369人参与
# 如何缓解入职前的焦虑 #
192364次浏览 1339人参与
# 研究所VS国企,该如何选 #
184846次浏览 1783人参与
# 你秋招想去哪些公司 #
22196次浏览 809人参与
# 担心入职之后被发现很菜怎么办 #
130800次浏览 775人参与
# 秋招结束之后的日子 #
75200次浏览 911人参与