From 04f02ade83faa8c64ec03d6d15bf40b5c926844a Mon Sep 17 00:00:00 2001 From: Akshata13-AI <71546169+Akshata13-AI@users.noreply.github.com> Date: Sun, 11 Oct 2020 14:46:38 +0530 Subject: [PATCH] Painting fence Algorithm program of implement the algorithm of painting fence in python --- PaintingfenceAlgo | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 PaintingfenceAlgo diff --git a/PaintingfenceAlgo b/PaintingfenceAlgo new file mode 100644 index 0000000..63699af --- /dev/null +++ b/PaintingfenceAlgo @@ -0,0 +1,24 @@ +# Python3 program for Painting Fence Algorithm +# optimised version + +# Returns count of ways to color k posts +def countWays(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] + +# Driver code +n = 3 +k = 2 +print(countWays(n, k)) + +