您的位置:首页 > 其它

LintCode-打劫房屋

2015-05-23 16:51 127 查看
假设你是一个专业的窃贼,准备沿着一条街打劫房屋。每个房子都存放着特定金额的钱。你面临的唯一约束条件是:相邻的房子装着相互联系的防盗系统,且 当相邻的两个房子同一天被打劫时,该系统会自动报警。

给定一个非负整数列表,表示每个房子中存放的钱, 算一算,如果今晚去打劫,你最多可以得到多少钱 在不触动报警装置的情况下。

样例

给定
[3, 8, 4]
,
返回
8
.

挑战

O(n) 时间复杂度 且 O(1) 存储。

分析:又是一个动态规划问题,我们设dp[i]表示到i为止且包括i为最后一个盗窃的房子得到的最大值,很明显取决于dp[i-2]和dp[i-3],和dp[i-4]无关了,因为取dp[i-4]的情况可以包含在dp[i-2]的情况下

代码:

class Solution {
public:
/**
* @param A: An array of non-negative integers.
* return: The maximum amount of money you can rob tonight
*/
long long houseRobber(vector<int> A) {
// write your code here
int n = A.size();
if(n==0)
return 0;
long long ret = A[0];
vector<long long> dp(3,0);
for(int i=0;i<n;i++)
{
long long temp = 0;
for(int j=0;j<2;j++)
{
temp = max(temp,dp[j]+A[i]);
}
ret = max(ret,temp);
dp.push_back(temp);
dp.erase(dp.begin());
}
return ret;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: