Number of People Aware of a Secret LeetCode solution in Python

1
class Solution:
2
def peopleAwareOfSecret(self, n: int, delay: int, forget: int) -> int:
3
table = [0] * (forget + 1)
4
table[1] = 1
5
days = 1
6
while days <= n - 1:
7
count = 0
8
for k in range(forget - 1, -1, -1):
9
if k + 1 > delay:
10
table[k + 1] = table[k]
11
count += table[k]
12
elif k + 1 <= delay:
13
table[k + 1] = table[k]
14
table[1] = count
15
days += 1
16
count = 0
17
for k in range(1, forget + 1):
18
count += table[k]
19
return count % (pow(10, 9) + 7)

0

WPM •0 •0

100%

ACC •0 •0

0s

TIME •0