力扣-动态规划-123 买卖股票的最佳时机Ⅱ

思路

1. dp数组定义：每一天有五个状态，分别是

• 0:不持有
• 1:第一次持有
• 2:第一次不持有、
• 3:第二次持有
• 4:第二次不持有

1. 递推公式：

   dp[i][0] = 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]); 

2. dp数组初始化：

   dp[0][0] = 0;
   dp[0][1] -= prices[0];
   dp[0][2] = 0;
   dp[0][3] -= prices[0];
   dp[0][4] = 0;

3. 遍历顺序：顺序
4. 时间复杂度：     

代码

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        //  0:不持有 1:第一次持有、2:第一次不持有、3:第二次持有、4:第二次不持有
        vector<vector<int>> dp(prices.size(), vector<int>(5, 0));
        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] = 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 max(dp[prices.size()-1][2], dp[prices.size()-1][4]);
    }
};