added House Robber

0ff5ec · 0ff5ec · commit ee8d8149e008 · 2019-12-14T13:08:41.000-08:00
diff --git a/rob.py b/rob.py
@@ -0,0 +1,19 @@
+class Solution(object):
+    def rob(self, nums):
+        """
+        :type nums: List[int]
+        :rtype: int
+        """
+        if len(nums) == 0:
+            return 0
+        if len(nums) == 1:
+            return nums[0]
+        if len(nums) == 2:
+            return max(nums)
+        profit = [0]*len(nums)
+        profit[0] = nums[0]
+        profit[1] = max(nums[0], nums[1])
+        for i in range(2,len(nums)):
+            profit[i] = max(profit[i-2] + nums[i], profit[i-1])
+        print profit
+        return profit[-1]
