来自Leecode上的一道easy题目《房屋盗贼》,题目可简述为:你是一个职业的盗贼,去偷沿街的一连串房子,每个房子都有一定数目的珠宝。唯一的限制就是不能偷窃相邻的两个房屋,这样会触发报警系统。然后最后的问题是求最大收益。用编程术语描述即是:给定一个数组,里面存储的非负整数,代表相应房间的珠宝数量,按照不能访问相邻元素的原则,求取访问的元素之和的最大值。比如测例:输入[1,2,3,10,5],输出12;输入[1,2,3,0,0,10,5],输出14。
有算法嗅觉的人马上知道这样的问题可以用“动态规划”来解决。动态规划:“动态规划算法通常基于一个递推公式及一个或多个初始状态。当前子问题的将由上一次子问题的解推出。”上面这句话指出了动态规划算法的适用范围和实现方法。所以,关键就是找出相应的递推关系式:
这里这道题则有:maxValue[i]=max(maxValue[i-2]+nums[i],maxValue[i-1])。maxValue存储了行走到每步(每个房子)当前的最优解(和最大),nums存储了每个房间的珠宝数量。
AC代码:
class Solution {
public:
int rob(vector<int>& nums) {
//dynamic programing
//recursion formular maxV[i]=max(maxV[i-2]+nums[i],maxV[i-1]),
//where maxV[i] is the maximum amount of money when robber go to the ith house
int n=nums.size();
if(n==0)
return 0;
else if(n==1)
return nums[0];
else if(n==2)
return max(nums[0],nums[1]);
vector<int> maxV;
maxV.push_back(nums[0]);
maxV.push_back(max(nums[1],maxV[0]));
for(int i=2;i<n;i++) {
maxV.push_back(max(maxV[i-2]+nums[i],maxV[i-1]));
}
return maxV[n-1];
}
};