问题描述:
给定一列非负数,找到不相邻的数的最大和。
解题思路:
声明一个数组dp,dp[i]表示到数组的第i个元素为止不相邻的数能组合成的最大和。比如,给定数列{2,7,1,9},dp[0]=2,到dp[1],因为不能选相邻的数,所以dp[1]=7。然后是dp[2],因为同样的原则,要比较1+dp[0]和dp[1]谁大,将大的数值赋给dp[2]。所以递推公式就是:dp[i] = max(dp[i-1],dp[i-2]+nums[i])
源代码如下:
class Solution {
public:
int rob(vector<int>& nums) {
if(nums.size()==0) return 0;
if(nums.size()==1) return nums[0];
vector<int> dp;
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(nums[i]+dp[i-2],dp[i-1]));
return dp.back();
}
};
public:
int rob(vector<int>& nums) {
if(nums.size()==0) return 0;
if(nums.size()==1) return nums[0];
vector<int> dp;
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(nums[i]+dp[i-2],dp[i-1]));
return dp.back();
}
};