Update 198_House_Robber.java

seanprashad · Apr 22, 2021 · c648044 · c648044
1 parent c4c7f38
commit c648044
Showing 1 changed file with 6 additions and 16 deletions.
diff --git a/Dynamic Programming/198_House_Robber.java b/Dynamic Programming/198_House_Robber.java
@@ -1,23 +1,13 @@
 class Solution {
     public int rob(int[] nums) {
-        if (nums == null || nums.length == 0) {
-            return 0;
-        }
-        if (nums.length == 1) {
-            return nums[0];
-        }
-        if (nums.length == 2) {
-            return Math.max(nums[0], nums[1]);
-        }
-
-        int prevHouse = 0, prevTwoHouses = 0, currHouse = 0;
+        int[] dp = new int[nums.length + 1];
+        dp[0] = 0;
+        dp[1] = nums[0];
 
-        for (int i = 0; i < nums.length; i++) {
-            currHouse = Math.max(prevHouse, prevTwoHouses + nums[i]);
-            prevTwoHouses = prevHouse;
-            prevHouse = currHouse;
+        for (int i = 1; i < nums.length; i++) {
+            dp[i + 1] = Math.max(dp[i], dp[i - 1] + nums[i]);
         }
 
-        return currHouse;
+        return dp[nums.length];
     }
 }