[리트코드] 198. House Robber, 213. House Robber II
198. House Robber
나의 풀이
/**
* @param {number[]} nums
* @return {number}
*/
var rob = function (nums) {
if (nums.length === 0) return 0;
if (nums.length === 1) return nums[0];
const dp = [nums[0], Math.max(nums[0], nums[1])];
for (let i = 2; i < nums.length; i++) {
dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
}
return dp[nums.length - 1];
};
console.log(rob([1, 2, 3, 1]));
213. House Robber II
나의 풀이
/**
* @param {number[]} nums
* @return {number}
*/
var rob = function (nums) {
if (nums.length === 0) return 0;
if (nums.length === 1) return nums[0];
const dp = [nums[0], nums[0]];
const dp2 = [0, nums[1]];
for (let i = 2; i < nums.length; i++) {
if (i !== nums.length - 1) {
dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
}
dp2[i] = Math.max(dp2[i - 2] + nums[i], dp2[i - 1]);
}
return Math.max(dp[nums.length - 2], dp2[nums.length - 1]);
};
console.log(rob([1, 2, 1, 1]));
0에서 시작하는 경우와 1에서 시작하는 경우를 각각 구해서 더 큰 값을 선택하였다.