这个题是动态规划的典型题,这题的其它解法很简单。既然是动态规划,很多情况下就是路径在几种状态之间来回变化,在这种情况下我们介绍一种更加通用的方法:状态图的方法,这种方法在很多的课程都有过介绍,同时也是编程的一种技巧。
状态图法有两点:1.状态,2.动作。我们要先寻找这个问题的动作,然后根据动作确定状态。
1.确定这个题的动作,偷和不偷
2.根据偷和不偷两种动作,我们可以确定有两个状态s0(偷之前的最大财富),s1(偷之后的最大财富)。
请看下面的状态转移图,然后根据这个转移图,我们再做详细讲解。
对于一个房间,如果这个房间的前一个房间被偷了,则这个房间就不能偷了(因此只有一种路径转移),而如果这个房间的前一个房间没有被偷,则这个房间可以选择偷,也可以选择不偷(因此有两种路径转移),上面的状态方程就是以这个原理画出了的,即如下解释:
对于“s1”,当"偷"动作发生后,下一个房间就变成了不能被偷了,即转到了"s1"状态,也就是说"s1"只能通过"s0"以“偷”动作转移。
对于 “s0”,即使下面这种情况,对于一个房间,如果它的前一个房间的前一个房间被偷了,那么它的前一个房间就不能偷,而对于当前房间可以偷,也可以不偷,偷和不偷取决于哪种情况下积累的财富更大。
即有下面方程
s1[i] = s0[i-1] + nums[i];
s0[i] = max(s1[i-1],s0[i-1]);
i:表示当前房间,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> s0(nums.size(),0);
vector<int> s1(nums.size(),0);
s1[0] = nums[0];
for(int i = 1; i < nums.size(); i++)
{
s1[i] = s0[i-1] + nums[i];
s0[i] = max(s1[i-1],s0[i-1]);
}
return max(s0[nums.size()-1],s1[nums.size()-1]);
}
};
注意:上面代码中s0,s1可以不用定义成容器的形式,用int就可以了,这样可以更加节省空间。
请读者按照这种方法自己去做leetcode309题,原理一样,状态图请见如下:
从题目中可以看出,不管哪一天,都只能是 buy 或者 sell 或者 cooldown(rest) 三种动作中的一种。