From f05d888a0b621ca4e99e2b0fb6e23c097006fe41 Mon Sep 17 00:00:00 2001 From: Santo Cariotti Date: Tue, 28 May 2024 10:29:13 +0200 Subject: Init --- progs/a654.py | 9 +++++++++ 1 file changed, 9 insertions(+) create mode 100644 progs/a654.py (limited to 'progs/a654.py') 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 -- cgit v1.2.3-18-g5258