[leetcode] #213 House Robber II Medium (medium)

时间:2024-04-23 04:28:46

原题链接

比子母题House Robber多了一个条件:偷了0以后,第n-1间房子不能偷。

转换思路为求偷盗【0,n-1)之间,以及【1,n)之间的最大值。

用两个DP,分别保存偷不偷第0间房的情况。

Runtime: 0 ms, faster than 100.00%

class Solution
{
public:
int rob(vector<int> &nums)
{
int res = ;
int len = nums.size();
if (len <= )
return res;
if (len == )
return nums[];
if (len == )
return (nums[] > nums[] ? nums[] : nums[]);
int dp1[len];
int dp2[len];
// rob 0
dp1[] = nums[];
dp1[] = max(nums[], dp1[]);
for (int i = ; i < len - ; i++)
{
dp1[i] = max(dp1[i - ] + nums[i], dp1[i - ]);
}
//rob 1
dp2[] = ;
dp2[] = nums[];
dp2[] = max(nums[], dp2[]);
for (int i = ; i < len; i++)
{
dp2[i] = max(dp2[i - ] + nums[i], dp2[i - ]);
}
return max(dp1[len - ], dp2[len - ]);
}
};