EASY/src/easy/PaintFence.java

fishercoder1534 · fishercoder1534 · commit 1fcbeb388e77 · 2016-08-29T21:51:41.000-07:00
diff --git a/EASY/src/easy/PaintFence.java b/EASY/src/easy/PaintFence.java
@@ -0,0 +1,15 @@
+package easy;
+
+public class PaintFence {
+    public int numWays(int n, int k) {
+        if(n == 0) return 0;
+        else if(n == 1) return k;
+        int sameColorCnt = k, diffColorCnt = k*(k-1);
+        for(int i = 2; i < n; i++){
+            int temp = diffColorCnt;
+            diffColorCnt = (diffColorCnt+sameColorCnt)*(k-1);
+            sameColorCnt = temp;
+        }
+        return sameColorCnt+diffColorCnt;
+    }
+}
