|
1 | 1 | package leetcode.dp |
2 | 2 |
|
3 | 3 | /** [198. House Robber](https://leetcode.com/problems/house-robber) */ |
4 | | -fun rob1(nums: IntArray): Int { |
5 | | - var cur = 0 |
| 4 | +fun robBottomUp(nums: IntArray): Int { |
| 5 | + if (nums.size == 1) { |
| 6 | + return nums[0] |
| 7 | + } |
| 8 | + val dp = IntArray(nums.size) { -1 } |
| 9 | + dp[0] = nums[0] |
| 10 | + dp[1] = maxOf(nums[0], nums[1]) |
| 11 | + for (i in 2..nums.lastIndex) { |
| 12 | + dp[i] = maxOf(dp[i-1], nums[i] + dp[i - 2]) |
| 13 | + } |
| 14 | + return dp.last(); |
| 15 | +} |
| 16 | + |
| 17 | +fun robTopDown(nums: IntArray, idx: Int = 0, memo: IntArray = IntArray(nums.size) { -1 }): Int = |
| 18 | + when { |
| 19 | + idx >= nums.lastIndex + 1 -> 0 |
| 20 | + memo[idx] != -1 -> memo[idx] |
| 21 | + else -> { |
| 22 | + val include = nums[idx] + robTopDown(nums, idx + 2, memo) |
| 23 | + val exclude = robTopDown(nums, idx + 1, memo) |
| 24 | + maxOf(include, exclude).also { memo[idx] = it } |
| 25 | + } |
| 26 | + } |
| 27 | + |
| 28 | +fun rob(nums: IntArray): Int { |
6 | 29 | var prev = 0 |
| 30 | + var prevPrev = 0 |
7 | 31 | for (num in nums) { |
8 | | - val tmp = cur |
9 | | - cur = maxOf(cur, num + prev) |
10 | | - prev = tmp |
| 32 | + val tmp = prev |
| 33 | + prev = maxOf(prev, num + prevPrev) |
| 34 | + prevPrev = tmp |
11 | 35 | } |
| 36 | + return prev |
| 37 | +} |
12 | 38 |
|
13 | | - return cur |
| 39 | +fun main() { |
| 40 | + println(robBottomUp(intArrayOf(1, 2, 3, 1))) // 4 |
| 41 | + println(robBottomUp(intArrayOf(2,7,9,3,1))) // 12 |
| 42 | + |
| 43 | + println(robTopDown(intArrayOf(1, 2, 3, 1))) // 4 |
| 44 | + println(robTopDown(intArrayOf(2,7,9,3,1))) // 12 |
14 | 45 | } |
0 commit comments