今天的感觉太麻烦
写昨天的
cur = 第i天有多少人知道秘密
mem[i] = 第i天有多少人刚知道秘密
window_sum = 现在这个时间点有多少人能传达秘密
然后就照做
看起来很直觉 我写起来跟智障一样
吐血
def peopleAwareOfSecret(self, n: int, delay: int, forget: int) -> int:
cur = 1
mem = [1]
mod = 10**9 + 7
window_sum = 0
for i in range(1, n):
if (i-forget) >= 0:
window_sum -= mem[i-forget]
cur -= mem[i-forget]
if (i-delay) >= 0:
window_sum += mem[i-delay]
cur = (cur + window_sum) % mod
mem.append(window_sum)
return cur