打家劫舍&&打家劫舍II

打家劫舍

题目

思路

使用动态规划解决，状态表示：dp[i]表示从0到 i 房屋偷盗到的最大金额。

状态转移方程：dp[i]=dp[i-1]（不偷盗 i 号房屋）

                         dp[i]=dp[i-2]+nums[i]（偷盗 i 号房屋）

                         即dp[i]=max(dp[i-1],dp[i-2]+nums[i])

代码

class Solution {
public:
    int dp[110];

    int rob(vector<int>& nums) {
        dp[0]=nums[0];
        if(nums.size()==1) return dp[0];
        dp[1]=max(nums[0],nums[1]);
        for(int i=2;i<nums.size();i++)
        {
            dp[i]=max(dp[i-1],dp[i-2]+nums[i]);
        }
        return dp[nums.size()-1];
    }
};

优化

结合状态转移方程可知，填写dp[i]只需要用到前两个位置的值，那么可以定义一个长度为2的数组来解决，空间复杂度为O（N）。

class Solution {
public:
    int dp[2];

    int rob(vector<int>& nums) {
        dp[0]=nums[0];
        if(nums.size()==1) return dp[0];
        dp[1]=max(nums[0],nums[1]);
        for(int i=2;i<nums.size();i++)
        {
            dp[i%2]=max(dp[(i+1)%2],dp[(i+2)%2]+nums[i]);
        }
        return dp[(nums.size()-1)%2];
    }
};

打家劫舍II

题目

思路

这道题区别于上边的题，在于房屋首尾相连，形成了环，需要格外注意的点就是偷盗了0号房屋，就不能偷盗最后一号房屋，偷盗了最后一号房屋，就不能偷盗0号房屋，解决了这两点，就可以部分复用之前的代码来解决了。

class Solution {
public:
    int dp[110];

    int rob(vector<int>& nums) {
        dp[0]=nums[0];
        if(nums.size()==1) return dp[0];
        int a=findmax(nums,0,nums.size()-2);
        memset(dp,0,sizeof dp);
        int b=findmax(nums,1,nums.size()-1);
        return max(a,b);
    }

    int findmax(vector<int>& nums,int l,int r){
        dp[0]=nums[l];
        if(r>l)
            dp[1]=max(nums[l],nums[l+1]);
        for(int i=2;i<=r-l;i++)
            dp[i]=max(dp[i-1],dp[i-2]+nums[l+i]);
        return dp[r-l];
    }
};

优化

结合状态转移方程，填写dp[i]时会用到前两个值，那么可以开辟2个空间，空间复杂度为O（N）。

class Solution {
public:
    int dp[2];

    int rob(vector<int>& nums) {
        dp[0]=nums[0];
        if(nums.size()==1) return dp[0];
        int a=findmax(nums,0,nums.size()-2);
        memset(dp,0,sizeof dp);
        int b=findmax(nums,1,nums.size()-1);
        return max(a,b);
    }

    int findmax(vector<int>& nums,int l,int r){
        dp[0]=nums[l];
        if(r>l)
            dp[1]=max(nums[l],nums[l+1]);
        for(int i=2;i<=r-l;i++)
            dp[i%2]=max(dp[(i-1)%2],dp[(i-2)%2]+nums[l+i]);
        return dp[(r-l)%2];
    }
};