diff --git a/PaintingfenceAlgo b/PaintingfenceAlgo new file mode 100644 index 00000000..63699afd --- /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)) + +