代码随想录训练营第五十天123.买卖股票的最佳时机III188.买卖股票的最佳时机IV

areyouapoemer2023-11-29 19:20

123.买卖股票的最佳时机III

题目链接 123. 买卖股票的最佳时机 III - 力扣（LeetCode）

讲解链接代码随想录 (programmercarl.com)

之前做过只有一次购买机会的股票题，但是一道了两次就不知道如何是好了，答案代码随想录 (programmercarl.com)

的方法确实朴实无华：每一天都有可能的五个状态：

0 从未购买过股票

1 第一次持有

2 第一次卖出

3 第二次持有

4 第二次卖出

外层for循环，内层每次更新这五个状态的值：

cpp 复制代码

        for(int i=1;i<prices.size();i++){
            dp[i][0] = dp[i - 1][0];
            dp[i][1]=max(dp[i-1][1],dp[i-1][0]-prices[i]);
            dp[i][2]=max(dp[i-1][1]+prices[i],dp[i-1][2]);
            dp[i][3]=max(dp[i-1][2]-prices[i],dp[i-1][3]);
            dp[i][4]=max(dp[i-1][3]+prices[i],dp[i-1][4]);
        }

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

题目链接 188. 买卖股票的最佳时机 IV - 力扣（LeetCode）

讲解链接代码随想录 (programmercarl.com)

相较于III，IV的购买次数变成了不确定，这时就需要总结规律：能取两次有五个状态，能取三次有七个状态，那么取k次就变成了2*k+1个状态，此时可以在天数的i循环中加入多个状态的j循环：

cpp 复制代码

        for(int i=1;i<prices.size();i++){
            for(int j=1;j<=k;j++){
                
                dp[i][2*j-1]=max(dp[i-1][2*j-1],dp[i-1][2*j-2]-prices[i]);
                dp[i][2*j]=max(dp[i-1][2*j],dp[i-1][2*j-1]+prices[i]);
            }
        }