开始循环前必须更新一下A[1]的取值

yingl · yingl · commit eaff977bd170 · 2016-09-26T12:13:55.000+08:00
diff --git a/house_robber.py b/house_robber.py
@@ -12,10 +12,12 @@ def houseRobber(self, A):
         '''
         if not A:
             return 0
-        if len(A) == 1:
+        elif len(A) == 1:
             return A[0]
-        if len(A) == 2:
+        elif len(A) == 2:
             return max(A[0], A[1])
-        for i in xrange(2, len(A)):
-            A[i] = max(A[i] + A[i - 2], A[i - 1])
-        return A[-1]
+        else:
+            A[1] = max(A[0], A[1])
+            for i in xrange(2, len(A)):
+                A[i] = max(A[i] + A[i - 2], A[i - 1])
+            return A[-1]
