原题链接:打家劫舍
个人解法
思路:
这是一个DP问题,每间房屋存在两种状态,要么被偷,要么没被偷。那么我们可以利用自动机直观的解决状态转移的问题,如果我们将两种状态视为状态机的两个结点(被偷:1, 没被偷:0),那么我们可以知道存在下列的边:
- 0 -> 0:表示下一家房间仍然不偷,状态不变
- 0 -> 1:下一家偷
- 1 -> 0:下一家只能不偷
那么由此我们其实就可以写出状态转移方程,此时状态方程为f[i][0]
表示前i间房间且最后一间房间为被偷的最大金额,同理f[i][1]
故状态转移方程:
f[i][1] = f[i - 1][0] + nums[i - 1];
f[i][0] = max(f[i - 1][0], f[i - 1][1]);
时间复杂度: O ( n ) O(n) O(n)
代码:
class Solution {
public:
int f[110][2];
int rob(vector<int>& nums) {
int n = nums.size();
f[1][0] = 0, f[1][1] = nums[0];
for(int i = 2;i <= n;i ++) {
f[i][1] = f[i - 1][0] + nums[i - 1];
f[i][0] = max(f[i - 1][0], f[i - 1][1]);
}
return max(f[n][0], f[n][1]);
}
};