第 1 步:状态定义
dp[i]:偷到i号屋时 能偷到的最大金额
第 2 步:推导状态转移方程
dp[i] = nums[i] + maxdp_i_2
其中 maxdp_i_2 = max(dp[i - 2], maxdp_i_2)
因为偷东西不一定必须只隔一间屋子,隔两间也可以,如果只是按隔一间屋子计算,那么样例2 1 1 2
就会出错,因为该例中隔2间屋子偷才能最大化。
所以状态转移方程里的第二部分应该是从dp[0]到dp[i-2]里的最大值M,从M的那间屋子开始偷才是最大化金额
第 3 步:思考初始化
前两间屋子的最大化数值就是自己本身:
dp[0] = nums[0]
dp[1] = nums[1]
由于我们是从第3间(下标2)屋子开始循环,所以dp[i-2]之前的最大值M就是第1间(下标0)屋子的金额:
maxdp_i_2 = dp[0]
第 4 步:思考输出
比较【以最后一间屋子和以倒数第二间屋子为终点时的金额,即dp[n]和dp[n-1]】,返回较大者
int rob(vector<int>& nums) {
//数组大小为0 1 2时可以单独处理
if (nums.size() == 0)
return 0;
else if (nums.size() == 1)
return nums[0];
else if (nums.size() == 2)
return max(nums[0], nums[1]);
int n = nums.size();
int* dp = new int[n];
// dp[i]:偷到i号屋时 能偷到的最大金额
dp[0] = nums[0];
dp[1] = nums[1];
int maxdp_i_2 = dp[0];
for (int i = 2; i < n; ++i)
{
maxdp_i_2 = max(dp[i - 2], maxdp_i_2);
dp[i] = nums[i] + maxdp_i_2;
}
return max(dp[n - 1], dp[n - 2]);
}