今天的感覺太麻煩
寫昨天的
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