欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

leetcode常见-打家劫舍系列

程序员文章站 2022-07-12 09:09:33
...
  1. leetcode-198-打家劫舍
    题型:动态规划
    难度:简单
    题目:你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
    给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
    代码:
class Solution {
//对于每间房子,都有两种可能,偷了或没偷
public:
    int rob(vector<int>& nums) {
        int n = nums.size();
        if(n == 0) return 0;
        if(n == 1) return nums[0];
        vector<vector<int> > dp(n,vector<int>(2,0));//0表示偷1表示没偷
        dp[0][0] = nums[0];
        dp[0][1] = 0;
        for(int i=1;i<n;i++)
        {
            //如果偷这间,那么上一间一定没偷
            dp[i][0] = dp[i-1][1]+nums[i];
            //如果不偷这间,上一次偷和没偷选最大
            dp[i][1] = max(dp[i-1][0],dp[i-1][1]);
        }
        return max(dp[n-1][0],dp[n-1][1]);
    }
};

空间过大,优化如下:

class Solution {
//对于每间房子,都有两种可能,偷了或没偷
public:
    int rob(vector<int>& nums) {
        int n = nums.size();
        if(n == 0) return 0;
        if(n == 1) return nums[0];
        vector<int> dp(n,0);
        dp[0] = nums[0];
        dp[1] = max(dp[0],nums[1]);
        for(int i=2;i<n;i++)
        {
            dp[i] = max(dp[i-2]+nums[i],dp[i-1]);
        }
        return dp[n-1];
    }
};
  1. leetcode-213-打家劫舍2
    题型:动态规划
    难度:中等
    题目:你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
    给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
    代码:
class Solution {
public:
    int rob(vector<int>& nums) {
        int n = nums.size();
        if(n == 0) return 0;
        if(n == 1) return nums[0];
        if(n == 2)
            return max(nums[0],nums[1]);
        //因为是环形,所以如果选第一家,一定不能偷最后一家;如果选最后一家,一定不能偷第一家。
        //1.偷第一家发
        vector<int> dp0(n-1,0);
        dp0[0] = nums[0];
        dp0[1] = max(nums[0],nums[1]);
        //对于dp0,范围是2~n-2间
        for(int i=2;i<n-1;i++)
        {
            dp0[i] = max(dp0[i-2]+nums[i],dp0[i-1]);
        }
        //2.不偷第一家
        vector<int> dp1(n,0);
        dp1[0] = nums[1];
        dp1[1] = max(nums[1],nums[2]);
        //对于dp1,范围是3~n-1间
        for(int i=2;i<n-1;i++)
        {
            dp1[i] = max(dp1[i-1],dp1[i-2]+nums[i+1] ); 
        }
        return max(dp0[n-2],dp1[n-2]);

    }
};
  1. leetcode-337-打家劫舍
    题型:动态规划、回溯
    难度:中等
    题目:在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为“根”。 除了“根”之外,每栋房子有且只有一个“父“房子与之相连。一番侦察之后,聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫,房屋将自动报警。
    计算在不触动警报的情况下,小偷一晚能够盗取的最高金额。
    代码:
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    unordered_map<TreeNode *,int> mp;
public:
    int rob(TreeNode* root) {
        if(root == nullptr) return 0;
        if(root->left==nullptr && root->right==nullptr)
            return root->val;
        if(mp[root])
            return mp[root];
        //如果偷当前节点
        int res1 = root->val;
        if(root->left)
        {
            res1 += rob(root->left->left)+rob(root->left->right);
        }
        if(root->right)
        {
            res1 += rob(root->right->left)+rob(root->right->right);
        }
        //如果不偷当前节点
        int res2 = 0;
        res2 = rob(root->left)+rob(root->right);
        mp[root] = max(res1,res2);
        return mp[root];
    }
};