[리트코드] 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에서 시작하는 경우를 각각 구해서 더 큰 값을 선택하였다.