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