Update House Robber.java

varunu28 · web-flow · commit eb1a80cf3392 · 2024-01-21T09:18:41.000-08:00
diff --git a/Medium/House Robber.java b/Medium/House Robber.java
@@ -1,14 +1,11 @@
 class Solution {
-  public int rob(int[] nums) {
-    if (nums.length == 1) {
-      return nums[0];
+    public int rob(int[] nums) {
+        int n = nums.length;
+        int[] dp = new int[n + 1];
+        dp[1] = nums[0];
+        for (int i = 1; i < n; i++) {
+            dp[i + 1] = Math.max(dp[i], dp[i - 1] + nums[i]);
+        }
+        return dp[n];
     }
-    int[] dp = new int[nums.length];
-    dp[0] = nums[0];
-    dp[1] = Math.max(nums[0], nums[1]);
-    for (int i = 2; i < nums.length; i++) {
-      dp[i] = Math.max(dp[i - 1], dp[i - 2] + nums[i]);
-    }
-    return dp[nums.length - 1];
-  }
 }
