Keshawn_lu's Blog

Leetcode 198. 打家劫舍

字数统计: 380阅读时长: 1 min
2020/05/29 Share

题目简介:

你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警

给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。

示例 1:

1
2
3
4
输入: [1,2,3,1]
输出: 4
解释: 偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
  偷窃到的最高金额 = 1 + 3 = 4 。

示例 2:

1
2
3
4
输入: [2,7,9,3,1]
输出: 12
解释: 偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
  偷窃到的最高金额 = 2 + 9 + 1 = 12 。

思路:

动态规划即可,状态方程为dp[i] = max(nums[i] + dp[i - 2], dp[i - 1]),需要注意的是dp[1]max(dp[0], dp[1])

代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
class Solution {
public:
int rob(vector<int>& nums) {

if(nums.empty())
return 0;

if(nums.size() < 2){

return nums[0];
}

vector<int> dp(nums.size(), 0);

dp[0] = nums[0];
dp[1] = nums[1] > nums[0] ? nums[1] : nums[0];


int max_res = dp[0] > dp[1] ? dp[0] : dp[1];

for(int i = 2; i < nums.size(); i++){

dp[i] = max(nums[i] + dp[i - 2], dp[i - 1]);

if(dp[i] > max_res)
max_res = dp[i];
}

return max_res;
}
};
CATALOG
  1. 1. 题目简介:
  2. 2. 思路:
  3. 3. 代码如下: