买卖股票的最 佳时机 IV【leetcode 188】

解答

#include <vector>
#include <algorithm>
using namespace std;
// @lc code=start
class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        int n = prices.size(); //先获取数组的长度
        if(n==0) return 0; //特殊处理

        // 设置一个三维数组 dp
        // 大小为dp[i][j][b]
        // i 表示天数，dp[i] 表示第 i 天的最大利润
        // j 表示最多交易次数，每次交易包含买入和卖出一个完整流程
        // 注意：【 j 表示最多交易次数，而不是实际交易次数，比如最多交易两次可能实际只交易一次】
        // b 表示当前是否持有股票，取值为 0 和 1
        // 其中 0 表示当前持有 0 份股票，即【不持有】股票
        // 而 1 表示当前持有 1 份股票，即【持有】股票
        vector<vector<vector<int>>> dp(n,vector<vector<int>>(k+1,vector<int>(2)));

        //边界初始化
        //从最大交易次数j=1开始遍历，j=0时没有交易，采用默认值为0即可
        for (int j = 1; j < k+1; j++)
        {
            //dp[0][j][0] 表示在第 0 天结束时，即收盘后，手上持有 0 份股票，且此时最多进行了 j 次交易的情况下可以获得的最大收益
            dp[0][j][0] = 0;

            // dp[0][j][1] 表示在第 0 天结束时，即收盘后，手上持有 1 份股票，且此时最多进行了 j 次交易的情况下可以获得的最大收益
            // 手上持有了 1 份股票，那肯定是执行了买入操作，然后又还没有卖出，那么钱都投入了股票中，利润就是负的，即为 -prices[0]
            dp[0][j][1] = -prices[0];
        }
        
        // 动态规划：自底向上，即从前向后遍历，从i=1天开始遍历
        for (int i = 1; i < n; i++)
        {
            //从最大交易次数j=1开始遍历，j=0时没有交易，采用默认值为0即可
            for (int j = 1; j < k+1; j++)
            {
                // 1、今天【不持有】股票
                // 第 i - 1 天【持有】股票，第 i 天卖出
                // 昨天【持有】股票，今天卖出
                // vs
                // 第 i - 1 天【不持有】股票，第 i 天不操作
                // 昨天【不持有】股票，今天不操作
                dp[i][j][0] = max(dp[i-1][j][0],dp[i-1][j][1]+prices[i]);

                // 2、今天【持有】股票
                // 第 i - 1 天【持有】股票，第 i 天不操作
                // 昨天【持有】股票，今天不操作
                // vs
                // 第 i - 1 天【不持有】股票，第 i 天买入
                // 昨天【不持有】股票，今天买入
                dp[i][j][1] = max(dp[i-1][j][1],dp[i-1][j-1][0]-prices[i]);
            }            
        }

        // 表示第 n - 1 天结束时，即收盘后，手上持有 0 份股票，且此时最多进行了 k 次交易的情况下可以获得的最大收益
        return dp[n-1][k][0];
    }
};