标签:dp
You are a professional robber planning to rob houses along a street. Each house has a certain amount of money stashed, the only constraint stopping you from robbing each of them is that adjacent houses have security system connected and it will automatically contact the police if two adjacent houses were broken into on the same night.
Given a list of non-negative integers representing the amount of money of each house, determine the maximum amount of money you can rob tonight without alerting the police.
这是一道比较简单的DP题,递推式是f(n)=max(f(n-1),f(n-2)+num[n]).
一开始写的是纯递归算法,结果TLE了。
class Solution {
public:
int rober(vector<int> &num,int n){
if(n==0) return 0;
else if(n==2) return max(num[0],num[1]);
else if(n==1) return num[0];
else{
return max(num[n-1]+rober(num,n-2),rober(num,n-1));
}
}
int rob(vector<int> &num) {
rober(num,num.size());
}
}
然后就是应该选择空间换取时间的方法,因为按照上面的方法,f(3)、f(4)、之类的会多次访问,不如用一个数组将它们记录下来。时间复杂度是O(n)
class Solution {
public:
int ans[10000];
int rob(vector<int> &num) {
int n=num.size(),i;
if(n==0) return 0;
else if(n==1) return num[0];
else if(n==2) return max(num[0],num[1]);
else{
ans[0] = num[0];
ans[1] = max(num[0],num[1]);
for(i=2;i<n;i++){
ans[i] = max(ans[i-1],num[i]+ans[i-2]);
}
return ans[n-1];
}
}
};
标签:dp
原文地址:http://blog.csdn.net/iboxty/article/details/44936973