代码随想录第四十八天
198.打家劫舍
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额
设dp[i]为抢劫0-i房间所得的最大金钱
dp[i]=max(dp[i-1],dp[i-2]+nums[i])
初始化dp[0]=nums[0],dp[1]=max(dp[0],dp[i-1])
遍历从前往后
代码如下:
int steal1(vector<int>& s)
{
vector<int>dp(s.size(), 0);
dp[0] = s[0];
dp[1] = max(s[0], s[1]);
for (int i=2;i<s.size();i++)
{
dp[i] = max(dp[i - 1], dp[i - 2] + s[i]);
}
return dp[s.size() - 1];
}
213.打家劫舍II
你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈 ,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警 。
给定一个代表每个房屋存放金额的非负整数数组,计算你 在不触动警报装置的情况下 ,能够偷窃到的最高金额。
需要根据取数组首元素还是尾元素进行分类,剩下的和前一题一样的方式
代码如下:
int stealII(vector<int>& s)
{
if (s.size() == 0)return 0;
if (s.size() == 1)return s[0];
if (s.size() == 2)return max(s[0], s[1]);
int result1 = partionSteal(s, 0, s.size() - 2);
int result2 = partionSteal(s, 1, s.size() - 1);
return max(result1, result2);
}
int partionSteal(vector<int>& s, int start, int end)
{
vector<int>dp(s.size(), 0);
dp[start] = s[start];
dp[start + 1] = max(s[start], s[start + 1]);
for (int i=start;i<=end;i++)
{
dp[i] = max(dp[i - 1], dp[i - 2] + s[i]);
}
return dp[end];
}
337.打家劫舍 III
在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为“根”。 除了“根”之外,每栋房子有且只有一个“父“房子与之相连。一番侦察之后,聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫,房屋将自动报警。
计算在不触动警报的情况下,小偷一晚能够盗取的最高金额。
本题递推不难,但是自己还是没有AC,看了卡哥代码后大受震撼,有一种恍然大悟的感觉
代码如下:
unordered_map<TreeNode*, int>umap;
int stealIII(TreeNode* root)
{
if (root == NULL)return 0;
if (umap.find(root) != umap.end())
return umap[root];
int val1 = root->val;
if (root->left != NULL)
{
val1 += stealIII(root->left->left) + stealIII(root->left->right);
}
if (root->right != NULL)
{
val1 += stealIII(root->right->left) + stealIII(root->right->right);
}
int val2 = stealIII(root->left) + stealIII(root->right);
int result = max(val1, val2);
umap[root] = result;
return result;
}