将题意精简地叙述,即为抢劫犯在一排房子中抢劫,房子i中的钱的总量为nums[i],且相邻的房子不能同时抢劫。问最终最多能抢到多少钱?
该问题可以用DP思想进行解决,开辟数组ret,ret[i]表示抢到第i栋房子时可以抢到钱的最大数,
可得到状态方程:ret[i] = max{ret[i-1], ret[i-2]+nums[i]}
针对ret[0]和ret[1]需要单独拿出来进行计算,代码如下:
class Solution {
public:
int rob(vector<int>& nums) {
int len = nums.size();
if(len == 0) return 0;
if(len == 1) return nums[0];
vector<int> ret(len,0);
ret[0] = nums[0];
ret[1] = max(nums[0],nums[1]);
for(int i=2; i < len; i++){
ret[i] = max(ret[i-1],ret[i-2]+nums[i]);
}
return ret[len - 1];
}
};