代码随想录第50天|123.买卖股票的最佳时机III，188.买卖股票的最佳时机IV

LeetCode123.买卖股票的最佳时机III

题目链接：123. 买卖股票的最佳时机 III - 力扣（LeetCode）

思路：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.size() == 0) return 0;
        //dp数组以及下标的含义
        //一天一共就有五个状态，
        //0.没有操作 （其实我们也可以不设置这个状态）
        //1.第一次持有股票
        //2.第一次不持有股票
        //3.第二次持有股票
        //4.第二次不持有股票
        vector<vector<int>> dp(prices.size(), vector<int> (5, 0));
        //dp[i][j]中 i表示第i天，j为 [0 - 4] 五个状态，dp[i][j]表示第i天状态j所剩最大现金。
        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];
    }
};

LeetCode188.买卖股票的最佳时机IV 

题目链接：188. 买卖股票的最佳时机 IV - 力扣（LeetCode）

思路：

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        if(prices.size() == 0) return 0;
        vector<vector<int>> dp(prices.size(), vector<int> (2 * k + 1, 0));
        //几笔交易就初始化2*交易次数，买入卖出
        for(int i = 1; i < 2 * k + 1; i += 2) {//奇数次买入
            dp[0][i] -= prices[0];
        }
        for(int i = 1; i < prices.size(); i++) {
            dp[i][0] = dp[i - 1][0];//递推公式，由之前的III得到,奇数买入，偶数卖出
            for(int j = 1; j < 2 * k + 1; j++) {
                if(j % 2 == 1) dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - 1] - prices[i]);
                if(j % 2 == 0) dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - 1] + prices[i]);
            }
        }
        return dp[prices.size() - 1][2 * k];
    }
};