leetcode 198. 打家劫舍 C++ 2021-07-01

class Solution {
public:
    int rob(vector<int>& nums) 
    {
        int n = nums.size();// n间房屋
        if (n == 0)
            return 0;
        if (n == 1)
            return nums[0];

        vector<int> dp(n, 0);//dp[i] 代表 0-i之间能投的最大量 记一下num[i]被偷没 设一stoleFlag (0:没偷i, 1:偷了i)
        dp[0] = nums[0];
        dp[1] = max(nums[0], nums[1]);
        bool stoleFlag = 0;
        if (dp[1] == nums[1])
            stoleFlag = 1;

        for (int i = 2; i < n; ++i)
        {
            if (stoleFlag == 0)//上一家没偷，这家直接偷
            {
                dp[i] = dp[i - 1] + nums[i];
                stoleFlag = 1;
            }
            else //上一家偷了
            {
                if (nums[i] > nums[i - 1])//这一家看比上一家大就偷这一家，东西还回上一家
                {
                    dp[i] = max(dp[i - 1] - nums[i - 1] + nums[i], dp[i - 2] + nums[i]);//dp[i] 与 dp[i - 1] dp[i - 2]有关
                    stoleFlag = 1;
                } 
                else
                {
                    dp[i] = dp[i - 1];
                    stoleFlag = 0;   
                }         
            }
            // cout << dp[i] << endl;
        }
        return dp[n - 1];

    }
};

//优化代码
class Solution {
public:
    int rob(vector<int>& nums) 
    {
        int n = nums.size();// n间房屋
        if (n == 0)
            return 0;
        if (n == 1)
            return nums[0];

        int shang_shang_jia = nums[0];//上上家可偷的最大值
        int shang_yi_jia = max(nums[0], nums[1]);//上一家可偷的最大值
        int ans = shang_yi_jia;
        
        for (int i = 2; i < n; ++i)
        {
            if (shang_yi_jia == shang_shang_jia)//上一家没偷，这家直接偷
                ans = shang_yi_jia + nums[i];
            else //上一家偷了，    这一家偷那就是shang_shang_jia + nums[i]， 不偷就是shang_yi_jia
                ans = max(shang_yi_jia , shang_shang_jia + nums[i]);  
            shang_shang_jia = shang_yi_jia;
            shang_yi_jia = ans;  
        }
        return ans;
    }
};

//再继续改点
class Solution {
public:
    int rob(vector<int>& nums) 
    {
        int n = nums.size();// n间房屋
        if (n == 0)
            return 0;
        if (n == 1)
            return nums[0];

        int shang_shang_jia = nums[0];//上上家可偷的最大值
        int shang_yi_jia = max(nums[0], nums[1]);//上一家可偷的最大值
        int ans = shang_yi_jia;
        
        for (int i = 2; i < n; ++i)
        {
            ans = max(shang_yi_jia , shang_shang_jia + nums[i]);  //管它上家偷没偷 反正只有这两情况
            shang_shang_jia = shang_yi_jia;
            shang_yi_jia = ans;  
        }
        return ans;
    }
};