170、【动态规划】leetcode ——188. 买卖股票的最佳时机 IV：二维数组+一维数组 （C++版本）

题目描述

原题链接：188. 买卖股票的最佳时机 IV

解题思路

（1）二维数组

思路与 123. 买卖股票的最佳时机 III（二维数组+一维数组） 相同，不同的是从2次交易次数，变为可以为k次交易次数。但处理逻辑还是与之前相同。

• dp[i][2 * k - 1] = 第i天第k次已持有股票时的最大总收益 = max(第i-1天第k次已持有股票时的最大总收益，第i天第k次已持有股票时的最大总收益（第i-1天第k - 1次不持有股票时的最大总收益 - prices[i]）)

• dp[i][2 * k] = 第i天第k次已不持有股票时的最大总收益 = max(第i-1天第k次已不持有股票时的最大总收益，第i天第k次已不持有股票时的最大总收益（第i-1天第k次已持有股票时的最大总收益 + prices[i]）)

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        if(prices.empty() || prices.size() == 1)                  return 0;
        int n = prices.size();
        vector<vector<int>> dp(n, vector<int>(2 * k + 1));
        for(int i = 1; i < 2 * k + 1; i += 2) {
            dp[0][i] = -prices[0];            
        }
        
        for(int i = 1; i < n; i++) {
            for(int j = 1; j < 2 * k + 1; j += 2) {
                dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - 1] - prices[i]);
                dp[i][j + 1] = max(dp[i - 1][j + 1], dp[i - 1][j] + prices[i]);
            }
        }

        return dp[n - 1][2*k];
    }
};

（2）一维数组

优化方式与 123. 买卖股票的最佳时机 III（二维数组+一维数组） 相同。

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        if(prices.empty() || prices.size() == 1)                  return 0;
        int n = prices.size();
        vector<int> dp(2 * k + 1);
        for(int i = 1; i < 2 * k + 1; i += 2) {
            dp[i] = -prices[0];            
        }
        
        for(int i = 1; i < n; i++) {
            for(int j = 1; j < 2 * k + 1; j += 2) {
                dp[j] = max(dp[j], dp[j - 1] - prices[i]);
                dp[j + 1] = max(dp[j + 1], dp[j] + prices[i]);
            }
        }

        return dp[2*k];
    }
};

参考文章：188. 买卖股票的最佳时机 IV