198.打家劫舍
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。题目
class Solution {
public:
int rob(vector<int>& nums) {
if(nums.size()==0){
return 0;
}
if(nums.size()==1){
return nums[0];
}
vector<int> dp(nums.size());
dp[0]=nums[0];
dp[1]=max(nums[0],nums[1]);
for(int i=2;i<nums.size();i++){
dp[i]=max(dp[i-2]+nums[i],dp[i-1]);
}
return dp[nums.size()-1];
}
};
213.打家劫舍II
你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈 ,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警 。
给定一个代表每个房屋存放金额的非负整数数组,计算你 在不触动警报装置的情况下 ,今晚能够偷窃到的最高金额。题目
class Solution {
public:
int rob(vector<int>& nums) {
if(nums.size()==0){
return 0;
}
if(nums.size()==1){
return nums[0];
}
int result1=robRange(nums,0,nums.size()-2);
int result2=robRange(nums,1,nums.size()-1);
return max(result1,result2);
}
int robRange(vector<int>& nums,int start,int end){
if(end==start){
return nums[start];
}
vector<int> dp(nums.size());
dp[start]=nums[start];
dp[start+1]=max(dp[start],nums[start+1]);
for(int i=start+2;i<=end;i++){
dp[i]=max(dp[i-2]+nums[i],dp[i-1]);
}
return dp[end];
}
};
337.打家劫舍 III
小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为 root 。
除了 root 之外,每栋房子有且只有一个“父“房子与之相连。一番侦察之后,聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果 两个直接相连的房子在同一天晚上被打劫 ,房屋将自动报警。
给定二叉树的 root 。返回 在不触动警报的情况下 ,小偷能够盗取的最高金额 。题目
class Solution {
public:
unordered_map<TreeNode*,int> umap;
int rob(TreeNode* root) {
if(root==NULL){
return 0;
}
if(root->left==NULL && root->right==NULL){
return root->val;
}
if(umap[root]){
return umap[root];
}
int val1=root->val;
//偷父节点
if(root->left){
val1+=rob(root->left->left)+rob(root->left->right);
}
if(root->right){
val1+=rob(root->right->left)+rob(root->right->right);
}
//不偷父节点
int val2=rob(root->left)+rob(root->right);
umap[root]=max(val1,val2);
return max(val1,val2);
}
};
动态规划的方法:
class Solution {
public:
int rob(TreeNode* root) {
vector<int> result=robTree(root);
return max(result[0],result[1]);
}
vector<int> robTree(TreeNode* root){
if(root==NULL){
return {0,0};
}
vector<int> left=robTree(root->left);
vector<int> right=robTree(root->right);
//偷父节点
int val1=root->val+left[0]+right[0];
//不偷父节点
int val2=max(left[0],left[1])+max(right[0],right[1]);
return {val2,val1};
}
};