[LeetCode] House Robber II 打家劫舍之二

最新推荐文章于 2020-05-16 02:42:55 发布

weixin_41157881

最新推荐文章于 2020-05-16 02:42:55 发布

阅读量99

点赞数

CC 4.0 BY-SA版权

分类专栏： C++

本文链接：https://blog.youkuaiyun.com/weixin_41157881/article/details/98398263

C++ 专栏收录该内容

200 篇文章

订阅专栏

本文介绍了一种解决打家劫舍问题的动态规划算法，通过两个函数RobberHouse和RobberHouseII实现，分别处理不考虑首尾相连和考虑首尾相连的情况。代码使用C++编写，通过递推公式dp[i] = max(vec[i] + dp[i-2], dp[i-1])来更新状态，最后返回dp数组的最后一个元素作为最大抢劫金额。

简单改一下打家劫舍的一的代码就行了
#include
#include
using namespace std;
int maxT(int a, int b)
{
if (a > b)
{
return a;
}
return b;
}
int RobberHouse(vector vec)
{
vector dp(vec.size(), 0);
dp[0] = vec[0];
dp[1] = maxT(vec[0], vec[1]);
for (int i = 2; i < vec.size(); i++)
{
dp[i] = maxT(vec[i] + dp[i - 2], dp[i - 1]);
}
return dp.back();
}
int RobberHouseII(vector vec)
{
vector vec1(vec.begin(), vec.end() - 1);
vector vec2(vec.begin() + 1, vec.end());
return maxT(RobberHouse(vec1), RobberHouse(vec2));
}
int main()
{

cout << RobberHouse({ 2,7,9,3,1 }) << endl;
cout << RobberHouseII({ 2,3,2 }) << endl;
system(“pause”);
return 0;
}