代码随想录第50天

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

题解

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<vector<int>> dp(prices.size(),vector<int>(4,0));
        dp[0][0] = //第一次不持有股票
        dp[0][1] = -prices[0];//第一次持有股票
        dp[0][2] = 0;//第二次不持有股票
        dp[0][3] = -prices[0];//第二次持有股票
        for(int i = 1;i < prices.size();i++)
        {
            dp[i][0] = max(dp[i - 1][0],dp[i - 1][1] + prices[i]);
            dp[i][1] = max(dp[i - 1][1],-prices[i]);
            dp[i][2] = max(dp[i - 1][2],dp[i - 1][3] + prices[i]);
            dp[i][3] = max(dp[i - 1][3],dp[i - 1][0] - prices[i]);
        }
        return max(dp[prices.size() - 1][0],dp[prices.size() - 1][2]);
    }
};

思路

①dp[i][j]:第i天状态j所剩最大现金。
共有四种状态：
0：第一次不持有股票
1：第一次持有股票
2：第二次不持有股票
3：第二次持有股票
②递推公式：
dp[i][0]=max(沿用第i-1天不持有的状态，持有股票后卖出股票)=max(dp[i - 1][0],dp[i - 1][1] + prices[i])
dp[i][1]=max(沿用第i-1天持有的状态，第一次买入股票)=max(dp[i - 1][0],-prices[i])
dp[i][2]=max(沿用第i-1天不持有的状态，第二次持有股票后卖出股票)=max(dp[i - 1][0],dp[i - 1][3] + prices[i])
dp[i][3]=max(沿用第i-1天不持有的状态，第一次卖出股票后再买入股票)=max(dp[i - 1][0],dp[i - 1][0] + prices[i])
③初始化
dp[0][0] = 0;(第0天没有操作)
dp[0][1] = -prices[0];(第0天第一次买入)
dp[0][2] = -prices[0];(可以认为第0天买入卖出后又买入)
dp[0][4] = 0;(可以认为第0天买入卖出后又买入又卖出)

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

题解

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        vector<vector<int>> dp(prices.size(),vector<int>(2 * k + 1,0));//奇数买入，偶数卖出
        for(int i = 1;i < 2 * k + 1 ;i += 2)
        {
            dp[0][i] = -prices[0];
        }
        for(int i = 1;i < prices.size();i++)
        {
            for(int j = 0;j < 2 * k;j += 2)
            {
                dp[i][j + 1] = max(dp[i - 1][j + 1],dp[i - 1][j] - prices[i]);
                dp[i][j + 2] = max(dp[i - 1][j + 2],dp[i - 1][j + 1] + prices[i]);
            }
        }

        return dp[prices.size() - 1][2 * k];
    }
};

思路

dp[i][j]:第i天状态j所剩最大现金。
共有(2 * k + 1)种状态：
0：不操作
1：第一次持有股票
2：第一次不持有股票
3：第二次持有股票
4：第二次不持有股票
…
j为奇数：持有
j为偶数：不持有

注意

若没有dp[i][0]（不操作）的状态，在后续中循环内部会较难理出公式