【Leetcode】动态规划-123. 买卖股票的最佳时机 III

【Leetcode】动态规划-123. 买卖股票的最佳时机 III

题目

给定一个数组，它的第 i 个元素是一支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

思路

• 最多买卖两次，可以买卖一次，也可以买卖两次 也可以不卖
• dp[i][j]表示第i天状态J所剩最大现金 j有五个状态
• j:0-4 0:没有操作 1：买过一次股票 2 卖出一次股票 3 第二次买入 4 ：第二次卖出
• 状态转移方程：
• dp[i][1] = max(dp[i - 1][1],dp[i - 1][0] - prices[i]) 第i天买入股票 dp[i - 1][0] - prices[i] 第i天没有操作dp[i][1] = dp[i - 1][1]
• dp[i][2] = max(dp[i - 1][1] + prices[i],dp[i - 1][2]) 第i天卖出股票 dp[i - 1][1] + prices[1] 或者沿用前一天状态dp[i - 1][2]
• 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])

代码

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        // dp[i][j]表示第i天状态J所剩最大现金  j有五个状态
        // j:0-4   0:没有操作   1：买过一次股票  2 卖出一次股票  3 第二次买入 4 ：第二次卖出
        // 状态转移方程：
        // dp[i][1] = max(dp[i - 1][1],dp[i - 1][0] - prices[i]) 第i天买入股票 dp[i - 1][0] - prices[i] 第i天没有操作dp[i][1] = dp[i - 1][1]
        // dp[i][2] = max(dp[i - 1][1] + prices[i],dp[i - 1][2]) 第i天卖出股票 dp[i - 1][1] + prices[1]  或者沿用前一天状态dp[i - 1][2] 
        // 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])
        if(prices.size() == 0)
        {
            return 0;
        }
        vector<vector<int>> dp(prices.size(),vector<int>(5,0));

        // dp数组初始化
        dp[0][0] = 0;
        dp[0][1] = -prices[0];
        dp[0][2] = 0;
        dp[0][3] = -prices[0];
        dp[0][4] = 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];

    }
};