diff options
| author | Santo Cariotti <santo@dcariotti.me> | 2024-05-28 10:29:13 +0200 |
|---|---|---|
| committer | Santo Cariotti <santo@dcariotti.me> | 2024-05-28 10:29:13 +0200 |
| commit | f05d888a0b621ca4e99e2b0fb6e23c097006fe41 (patch) | |
| tree | eebbb2489144112d3288393e354d19375a0aa088 /progs/a654.py | |
Init
Diffstat (limited to 'progs/a654.py')
| -rw-r--r-- | progs/a654.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/progs/a654.py b/progs/a654.py new file mode 100644 index 0000000..8b15e73 --- /dev/null +++ b/progs/a654.py @@ -0,0 +1,9 @@ +def count_no_of_ways(n, k):
+ dp = [0] * (n + 1)
+ total = k
+ mod = 1000000007
+ dp[1] = k
+ dp[2] = k * k
+ for i in range(3,n+1):
+ dp[i] = ((k - 1) * (dp[i - 1] + dp[i - 2])) % mod
+ return dp[n]
\ No newline at end of file |