【代码随想录算法训练Day46】LeetCode 121.买卖股票的最佳时机、LeetCode 122.买卖股票的最佳时机II、LeetCode 123.买卖股票的最佳时机III

Day46 动态规划第八天

LeetCode 121.买卖股票的最佳时机

dp数组含义：dp[i][0]表示持有这支股票可获得的最大金额，dp[i][1]表示不持有这支股票可获得的最大金额
递推公式： dp[i][0]=max(dp[i-1][0],-price[i])
dp[i][1]=max(dp[i-1][1],dp[i-1][0]+price[i])
初始化：dp[0][0]=-price[0],dp[0][1]=0.
遍历顺序：从前往后遍历

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len=prices.size();
        if(len==0) return 0;
        vector<vector<int>> dp(len,vector<int>(2));
        dp[0][0]-=prices[0];
        dp[0][1]=0;
        for(int i=1;i<len;i++){
            dp[i][0]=max(dp[i-1][0],-prices[i]);
            dp[i][1]=max(dp[i-1][1],dp[i-1][0]+prices[i]);
        }
        return dp[len-1][1];
    }
};

LeetCode 122.买卖股票的最佳时机II

与上一题几乎相同，区别在于我们持有股票的话，卖出的金额不是0，而是dp[i-1][1]。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len=prices.size();
        if(len==0) return 0;
        vector<vector<int>> dp(len,vector<int>(2));
        dp[0][0]-=prices[0];
        dp[0][1]=0;
        for(int i=1;i<len;i++){
            dp[i][0]=max(dp[i-1][0],dp[i-1][1]-prices[i]);
            dp[i][1]=max(dp[i-1][1],dp[i-1][0]+prices[i]);
        }
        return dp[len-1][1];
    }
};

LeetCode 123.买卖股票的最佳时机III

dp数组含义：dp[i][0]不操作、dp[i][1]第一次持有、dp[i][2]第一次不持有、dp[i][3]第二次持有、dp[i][4]第二次不持有
递推公式：dp[i][0]=dp[i-1][0]、
dp[i][1]=max(dp[i-1][1],dp[i-1][0]-price[i])、
dp[i][2]=max(dp[i-1][2],dp[i-1][1]+price[i])、
dp[i][3]=max(dp[i-1][3],dp[i-1][2]-price[i])、
dp[i][4]=max(dp[i-1][4],dp[i-1][3]+price[i])
初始化：dp[0][0]=0,dp[0][1]=-price[0],dp[0][2]=,dp[0][3]=-price[0],dp[0][4]=0
遍历顺序：从小到大

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.size()==0) return 0;
        vector<vector<int>> dp(prices.size(),vector<int>(5,0));
        dp[0][1]=-prices[0];
        dp[0][3]=-prices[0];
        for(int i=1;i<prices.size();i++){
            dp[i][0]=dp[i-1][0];
            dp[i][1]=max(dp[i-1][1],dp[i-1][0]-prices[i]);
            dp[i][2]=max(dp[i-1][2],dp[i-1][1]+prices[i]);
            dp[i][3]=max(dp[i-1][3],dp[i-1][2]-prices[i]);
            dp[i][4]=max(dp[i-1][4],dp[i-1][3]+prices[i]);
        }
        return dp[prices.size()-1][4];
    }
};

要好好考虑一下，才能把递推的条件搞清楚。