1 Star 0 Fork 0

MetaverseMobile/leetcode

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
213_打家劫舍II.cpp 1.72 KB
一键复制 编辑 原始数据 按行查看 历史
TieNan2019 提交于 2020-10-28 22:46 +08:00 . Create 213_打家劫舍II.cpp
class Solution {
public:
map<bool, map<int, int>> m;
int recursion(const vector<int> nums,
int index = 0, bool head = false) {
// 超出范围
if (index >= nums.size())
return 0;
// 头节点已经被占用,且到达末尾节点
if (index == nums.size() - 1 && head)
return 0;
if (index == nums.size() - 1 && !head)
return nums[index];
// int steal = nums[index] + recursion(nums, index+2, head);
int steal = nums[index];
if (m[head].find(index+2) == m[head].end()) {
int r = recursion(nums, index+2, head);
m[head][index+2] = r;
steal += r;
}
else {
steal += m[head][index + 2];
}
// int keep = recursion(nums, index+1, head);
int keep = 0;
if (m[head].find(index+1) == m[head].end()) {
int k = recursion(nums, index+1, head);
m[head][index+1] = k;
keep = k;
}
keep = m[head][index+1];
return steal > keep ? steal : keep;
}
int rob(vector<int>& nums) {
m[true][0] = nums.front();
int head = nums.front() + recursion(nums, 2, true);
int noHead = recursion(nums, 1, false);
// cout << head << endl;
// cout << noHead << endl;
return head > noHead ? head : noHead;
}
};
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C++
1
https://gitee.com/MetaverseMobile/leetcode.git
git@gitee.com:MetaverseMobile/leetcode.git
MetaverseMobile
leetcode
leetcode
main

搜索帮助

371d5123 14472233 46e8bd33 14472233