(Leetcode) 198 - House Robber
기회가 되어 달레님의 스터디에 참여하여 시간이 될 때마다 한문제씩 풀어보고 있다.
https://leetcode.com/problems/house-robber/
풀어본 적이 없는데 풀어본 것 같은 느낌이 들 정도로 전형적인 dp 문제 형태이다.
dp array를 이용해서
- 0 번째 집을 방문할 경우 최대로 얻을 수 있는 값
- 1 번째 집을 방문할 경우 최대로 얻을 수 있는 값
- 2 번째 집을 방문할 경우 최대로 얻을 수 있는 값
- 3 번째 집을 방문할 경우 최대로 얻을 수 있는 값
- …
- n 번째 집을 방문할 경우 최대로 얻을 수 있는 값
이런식으로 계속 계산해 나간다.
n이 2보다 클 때 n 번째 집을 방문할 경우 최대로 얻을 수 있는 값은 Math.max(nums[i] + dp[i - 2], nums[i] + dp[i - 3]) 이다. 이는 연속된 집을 방문하지 못한다는 특성에 의한 것이다.
내가 작성한 풀이
public class Solution {
public int rob(int[] nums) {
int[] dp = new int[nums.length];
int max = 0;
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] + nums[0], nums[1]);
}
dp[0] = nums[0];
dp[1] = nums[1];
dp[2] = nums[2] + nums[0];
max = Math.max(dp[2], dp[1]);
for (int i = 3; i < nums.length; 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) 이다.