LeetCode-188. 买卖股票的最佳时机 IV

传送门：https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-iv/

思路：DP, 选对一个好的DP状态很重要啊。。。
dp[i][k][0]: 表示在前 i 天内最多参与 k 笔交易后手里没有购买股票的最大价值
dp[i][k][1]: 表示在前 i 天内最多参与 k 笔交易后手里有购买股票的最大价值，此时手里购买股票的那次也算在k笔交易之中
确定好dp状态,则状态转移方程就比较明显了,对于当前状态，要么是保存不变，要么是购买或出售股票
dp[i][k][0]=max(dp[i-1][k][0],dp[i-1][k][1]+prices[i-1]);
dp[i][k][1]=max(dp[i-1][k][1],dp[i-1][k-1][0]-prices[i-1]);
最后记得注意下初始状态的值即可

Code:

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