dp问题
找一下规律,使用数学归纳法
数组有一个元素那么就返回第一个,数组有两个元素,返回二者最大的,
数组有3个的话,返回第二个和低三个与第一个和的最大值。
用一个数组存一下dp
dp[0] = n[0];
dp[1] = n[1];
dp[2] = max(dp[0] + n[2], dp[1])
…
dp[n] = max(dp[n-2] + n[n], dp[n-1])
那么就很容易写出代码:
// 198. House Robber
int Solution::rob(vector<int> &nums)
{
vector<int> dp;
if (nums.size() == 0)
return 0;
if (nums.size() == 1)
return nums[0];
if (nums.size() == 2)
return max(nums[0], nums[1]);
dp.push_back(nums[0]);
dp.push_back(max(nums[0], nums[1]));
for (int i = 2; i < nums.size(); ++i)
{
dp.push_back(max(dp[i - 2] + nums[i], dp[i - 1]));
}
return dp[nums.size()-1];
}
Runtime: 4 ms, faster than 100.00% of C++ online submissions for House Robber.
Memory Usage: 9.2 MB, less than 7.95% of C++ online submissions for House Robber.