Skip to content

Commit 6c9c4ab

Browse files
committed
solve: house robber
1 parent 56b1d5c commit 6c9c4ab

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

house-robber/JustHm.swift

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
class Solution {
2+
func rob(_ nums: [Int]) -> Int {
3+
guard nums.count != 1 else { return nums[0] }
4+
guard nums.count != 2 else { return max(nums[0], nums[1]) }
5+
6+
// var dp = [nums[0], max(nums[0], nums[1])]
7+
var twoStepPrev = nums[0]
8+
var oneStepPrev = max(nums[0], nums[1])
9+
for i in 2..<nums.count {
10+
var maxNum = max(oneStepPrev, twoStepPrev + nums[i])
11+
twoStepPrev = oneStepPrev
12+
oneStepPrev = maxNum
13+
// var maxNum = max(dp[i-1], dp[i-2] + nums[i])
14+
// dp.append(maxNum)
15+
}
16+
// print(dp)
17+
// return dp.max() ?? 0
18+
return oneStepPrev
19+
}
20+
}

0 commit comments

Comments
 (0)