问题描述
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/house-robber
输入
[1,2,3,1]
输出
4
问题思想
每个房子的价值用H数组表示,H0表示第一间房子的价值。用S表示偷前几间房子的最大价值,S0表示偷第一间房子的最大价值,S2表示偷前三间房子的最大价值。
偷第一间屋S0=H0=1;
偷前两间屋S1=max{H1,H0}=2;
偷前三间屋(是否偷第三间):S2=max{S1,S0+H2};
- 如果偷第三间,S2=S0+H2;
- 不偷第三间,S2=S1
所以S2=max{S1,S0+H2}=4
偷前四间屋(是否偷第四间):S3=max{S2,S1+H3}
- 如果偷第三间,S3=S1+H2;
- 不偷第三间,S2=S2
所以有递推关系式:设房子个数为s[0]表示偷前一个房子的价值
当n=0时,s=0;
当n=0时,s[0]=h[0];
当n=1时,s[1]=max{h[0],h[1]}
当n>=2时,s[n]=max{h[n-1],s[n-2]+H[n]}
代码实现
#include"iostream"
#include<vector>
#include<algorithm>
using namespace std;
vector<int> states;
int stole(vector<int> &h)
{
int n;int i;
n = sizeof(h) / sizeof(int);
if (n == 0)
return 0;
if (n == 1)
return h[0];
states.push_back(h[0]);//为states[0]赋值
states.push_back(max(h[0], h[1]));//为states[1]赋值
for (i = 2;i < n;i++)
{
states.push_back(max(states[i - 2] + h[i], states[i - 1]));
}
return states[n - 1];
}
int main()
{
int n;
cin >> n;
vector<int> h(n);
for (int i = 0;i < n;i++)
{
cin >> h[i];
}
cout << stole(h) << endl;
}
优化空间复杂度,每次只用到H[n-1]与H[n-2]!!!
就是用first,second不断更新states[i-2]与states[i-1]的值;将空间复杂度由O(n)变为O(1);
#include"iostream"
#include<vector>
#include<algorithm>
using namespace std;
int stole(vector<int> &h)
{
int n;int i;
n = sizeof(h) / sizeof(int);
if (n == 0)
return 0;
if (n == 1)
return h[0];
int first = h[0];//为states[i-2]赋值
int second=max(h[0], h[1]);//为states[i-1]赋值,最大代价
for (i = 2;i < n;i++)
{
int temp = second;
second=max(first + h[i], second);//最新计算的代价值作为新的second,原来的second作为新first
first = temp;
}
return second;
}
int main()
{
int n;
cin >> n;
vector<int> h(n);
for (int i = 0;i < n;i++)
{
cin >> h[i];
}
cout << stole(h) << endl;
}