题目
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
示例1
输入: [1,2,3,1]
输出: 4
解释: 偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。
示例2
输入: [2,7,9,3,1]
输出: 12
解释: 偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
偷窃到的最高金额 = 2 + 9 + 1 = 12 。
思路
- 每一天都有两种选择,偷还是不偷。
- 如果今天要偷,那就得昨天没有偷。
- 如果今天没偷,那么昨天可能偷了也可能没偷。
- 使用动态规划:
- d p [ i ] [ 0 ] dp[i][0] dp[i][0]表示第i天没偷东西,获得的最大利益。第i天没偷,那么就取第i-1天偷与没偷中较大的收益作为第i天没偷取得的最大收益。
- d p [ i ] [ 1 ] dp[i][1] dp[i][1]表示第i天偷了东西,获得的最大利益。第i天能偷,那么肯定第i-1天没偷,所以为 d p [ i ] [ i − 1 ] + n u m [ i ] dp[i][i-1]+num[i] dp[i][i−1]+num[i],即第i-i没偷获得的最大收益加上第i天偷的收益。
代码
class Solution {
public:
int rob(vector<int>& nums) {
if ( nums.size() == 0 ) return 0;
int size = nums.size();
vector<vector<int>> dp( size, vector<int>( 2, 0 ) );
dp[0][1] = nums[0];
for ( int i = 1; i < nums.size(); ++i ) {
dp[i][0] = max( dp[i-1][0], dp[i-1][1] );
dp[i][1] = dp[i-1][0] + nums[i];
}
return max( dp[size-1][0], dp[size-1][1] );
}
};