박종훈 알고리즘 블로그

(Leetcode) 213 - House Robber ii

기회가 되어 달레님의 스터디에 참여하여 시간이 될 때마다 한문제씩 풀어보고 있다.

https://neetcode.io/practice


https://leetcode.com/problems/house-robber-ii/

바로 전에 풀었던 House Robber 에서 확장된 문제이다.

원형 array를 사용해서 좀 더 고민을 해줘야 한다.

포인트는 첫번째 것을 선택하면 마지막 것을 선택하지 못한다는 것이다.

그래서 다음과 같이 두가지 케이스로 나눠서 계산 한 후 둘 중 더 큰 값을 반환하도록 하였다.

  • 첫번째 부터 n-1 번째까지
  • 두번째 부터 n 번째까지

내가 작성한 풀이

public class Solution {
    public int rob(int[] nums) {
        if (nums.length == 1) {
            return nums[0];
        }

        if (nums.length == 2) {
            return Math.max(nums[0], nums[1]);
        }

        if (nums.length == 3) {
            return Math.max(nums[2], Math.max(nums[0], nums[1]));
        }


        return Math.max(getMaxInRange(nums, 0, nums.length - 1), getMaxInRange(nums, 1, nums.length));
    }

    public int getMaxInRange(int[] nums, int start, int end) {
        int[] dp = new int[nums.length];
        int max;
        dp[start] = nums[start];
        dp[start + 1] = nums[start + 1];
        dp[start + 2] = nums[start + 2] + nums[start];
        max = Math.max(dp[start + 2], dp[start + 1]);
        for (int i = start + 3; i < end; i++) {
            dp[i] = Math.max(nums[i] + dp[i - 2], nums[i] + dp[i - 3]);
            max = Math.max(max, dp[i]);
        }
        return max;
    }
}

TC, SC

시간 복잡도는 O(n), 공간 복잡도는 O(n) 이다.