Leetcode C++《熱題 Hot 100-66》309. 最佳買賣股票時機含冷凍期

Leetcode C++《熱題 Hot 100-66》309. 最佳買賣股票時機含冷凍期

1. 題目

給定一個整數數組，其中第 i 個元素代表了第 i 天的股票價格 。​

設計一個算法計算出最大利潤。在滿足以下約束條件下，你可以儘可能地完成更多的交易（多次買賣一支股票）:

你不能同時參與多筆交易（你必須在再次購買前出售掉之前的股票）。
賣出股票後，你無法在第二天買入股票 (即冷凍期爲 1 天)。
示例:

輸入: [1,2,3,0,2]
輸出: 3
解釋: 對應的交易狀態爲: [買入, 賣出, 冷凍期, 買入, 賣出]

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-cooldown
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

2. 思路

• dp[i][j] j = 0 不持股 j = 1 持股 j=2 冷凍期 收益買入就-，賣出就+
• 動態規劃系列題目

3. 代碼

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        //
        int n = prices.size();
        if (n == 0)
            return 0;
        int **dp = new int*[n];
        for (int i = 0; i < n; i++)
            dp[i] = new int[3];
        //dp[i][j] j = 0 不持股 j = 1 持股 j=2 冷凍期  收益買入就-，賣出就+
        dp[0][0] = 0;
        dp[0][1] = - prices[0];
        dp[0][2] = 0;
        for (int i = 1; i < n; 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][2]-prices[i]);
            dp[i][2] = dp[i-1][0];
        }
        return max(max(dp[n-1][0], dp[n-1][1]), dp[n-1][2]);
    }
};