yan chuanli há 2 anos atrás
pai
commit
acee441d0a
1 ficheiros alterados com 12 adições e 2 exclusões
  1. 12 2
      climb_stairs.py

+ 12 - 2
climb_stairs.py

@@ -18,9 +18,19 @@ def f(n):
         if n == 2:
             return 2
         else:
-            if g(n - 1) in cach[n - 1]:
-                return cach[n - 1] + g(n)
+            if n-1 in cach:
+                f1 = cach[n-1]
             else:
+                f1 = g(n-1)
+                cach[n-1] = f1
+            if n-2 in cach:
+                f2 = cach[n-2]
+            else:
+                f2 = g(n-2)
+                cach[n-2] = f2
+        return f1 + f2
+    return g(n)
 
+print(f(4))