leetcode188 买卖股票最佳时机IV

题目

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

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

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

示例 1:

输入: [2,4,1], k = 2
输出: 2
解释: 在第 1 天 (股票价格 = 2) 的时候买入，在第 2 天 (股票价格 = 4) 的时候卖出，这笔交易所能获得利润 = 4-2 = 2 。
示例 2:

输入: [3,2,6,5,0,3], k = 2
输出: 7
解释: 在第 2 天 (股票价格 = 2) 的时候买入，在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。
随后，在第 5 天 (股票价格 = 0) 的时候买入，在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。

分析

不说了，我只能做出股票1,2题和手续费题，看看人家大神的方法，膜拜就是了。。。
一个通用方法团灭6道股票问题

代码

class Solution {
public:
    int maxProfit(int k_max, vector<int>& prices) {

        if (prices.empty()){
            return 0;
        }

        int len = prices.size();

        if (k_max > len / 2){
            return maxProfit_infinity(prices);
        }

        vector<vector<vector<int>>> dp(len, vector<vector<int>>(k_max + 1, vector<int>(2, 0)));

        for (int i = 0; i < len; i++){
            for (int k = 1; k <= k_max; k++){
                if (i == 0){
                    dp[i][k][0] = 0;
                    dp[i][k][1] = -prices[i];
                }else{
                    dp[i][k][0] = max(dp[i - 1][k][0], dp[i - 1][k][1] + prices[i]);
                    dp[i][k][1] = max(dp[i - 1][k][1], dp[i - 1][k - 1][0] - prices[i]);
                }
            }
        }

        return dp[len - 1][k_max][0];
    }

    int maxProfit_infinity(vector<int>& prices){
        int cur_min = INT_MAX, cur_profit = 0, res = 0;

        for (auto x : prices){
            if (x - cur_min > cur_profit){
                cur_profit = x - cur_min;
            }else{
                res += cur_profit;
                cur_profit = 0;
                cur_min = x;
            }
        }

        return res + cur_profit;
    }
};