关注
第三题(最多喝多少天咖啡): #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))
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
2025-12-30 17:58
Conservatoire National Supérieur Musique et Dance de Lyon Java 喵_coding:项目太烂了外卖+点评啊 而且寒假实习差不多到时候了 hc没多少了 要实在想要找那只能投投大厂试试了
点赞 评论 收藏
分享
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 在大厂上班是一种什么样的体验 #
11390次浏览 147人参与
# 你的mentor是什么样的人? #
50932次浏览 716人参与
# 程序员找工作至少要刷多少题? #
19771次浏览 255人参与
# 我和mentor的爱恨情仇 #
106068次浏览 942人参与
# 论秋招对个人心气的改变 #
12051次浏览 168人参与
# 机械人避雷的岗位/公司 #
43748次浏览 303人参与
# 校招第一份工作你干了多久? #
136552次浏览 597人参与
# 为了减少AI幻觉,你注入过哪些设定? #
5256次浏览 162人参与
# 设计人如何选offer #
189573次浏览 868人参与
# 你的秋招进行到哪一步了 #
2530193次浏览 23253人参与
# 机械人还在等华为开奖吗? #
312044次浏览 1582人参与
# 牛客AI体验站 #
7165次浏览 195人参与
# 重来一次,我还会选择这个专业吗 #
411131次浏览 3898人参与
# 我现在比当时_,你想录用我吗 #
9070次浏览 116人参与
# 12306一秒售罄,你抢到回家的票了吗? #
2136次浏览 50人参与
# 一张图晒一下你的AI员工 #
5414次浏览 121人参与
# AI Coding的使用心得 #
4900次浏览 105人参与
# 刚入职的你踩过哪些坑 #
7191次浏览 130人参与
# 关于春招/暑期实习,你想知道哪些信息? #
7864次浏览 121人参与
# 晒晒你司的新年福利 #
8766次浏览 108人参与