动态规划
相同的代码跑两遍,和(一)思路一样
f[i] 表示偷第 i 个房间,g[i] 表示不偷第 i 个房间
这里分两种情况:
-
不选起点,答案就是 res = max(f[n], g[n]) 最大值
-
选起点,那么就不能选终点,所以答案就是 res = max(res, g[n])
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param nums int整型vector
* @return int整型
*/
int rob(vector<int>& nums) {
int n = nums.size();
vector<int> f(n + 1), g(n + 1);
// 1. 不选起点
f[0] = 0, f[1] = 0;
g[0] = 0, g[1] = 0;
for (int i = 2; i <= n; i ++ ) {
f[i] = g[i - 1] + nums[i - 1];
// 如果不选 i,那么 i - 1 可选可不选
g[i] = max(f[i - 1], g[i - 1]);
}
int res = max(f[n], g[n]);
// 2. 选起点,那么就不能选终点
f[0] = 0, f[1] = nums[0];
g[0] = 0, g[1] = 0;
for (int i = 2; i <= n; i ++ ) {
f[i] = g[i - 1] + nums[i - 1];
g[i] = max(f[i - 1], g[i - 1]);
}
return max(res, g[n]);
}
};