LeetCode : Best Time to Buy and Sell Stock III & IV

首先让我们来考虑一下III的DP做法

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int buy1 = INT_MIN,buy2 = INT_MIN;
        int sell1 = 0, sell2 = 0;
        for(int i=0;i<prices.size();i++){
            sell2 = max(sell2,buy2+prices[i]);
            buy2 = max(buy2,sell1-prices[i]);
            sell1 = max(sell1,buy1+prices[i]);
            buy1 = max(buy1,-prices[i]);
        }
        return sell2;
    }
};

我们认为，初始时，股票的账面总额为0，buy1表示第一次买股票时账面总额，sell1表示第一次卖股票时的账面总额，每一次买或卖股票之后，账面总额都需要根据本次买或卖的价格进行更新，比如第8行计算buy2, 先看sell1-prices[i]这个计算式, 它表示拿第一次卖完之后的收益sell1减去本次买股票的价钱prices[i]，buy2的值等于它和上一次上一次得到的buy2最大值。另外计算buy1,sell1,sell2的逻辑是一样的。

求解完III之后，再来考虑IV，很容易想到需要在III的解法的for循环里加一个k步的循环，但是提交之后，发现报runtime error，经过查看测试用例，发现其中一个测试用例的k值是100000000，我去，搞了个这么大的K，这下我也是卡住了，不知道要怎么改进，最后无耻的百度之后，发现可以单独对此情况进行考虑，当k值大于prices的数量时，可以把问题等同于题目II的无数次买卖股票的情况，这样考虑之后对代码进行一下改进，终于AC了，完整代码如下：

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        if(k>=prices.size()){//当尝试次数超过prices的个数时，问题可以转换成了无数次买卖的情况。
            int maxP=0;
            for(int i=1;i<prices.size();i++){
                if(prices[i]>prices[i-1]){
                    maxP = maxP + (prices[i]-prices[i-1]);
                }
            }
            return maxP;
        }
        int* sell = new int[k];
        int* buy = new int[k];
        for(int i=0;i<k;i++){
            buy[i] = INT_MIN;
            sell[i] = 0;
        }
        for(int i=0;i<prices.size();i++){
            for(int j=(k-1);j>=0;j--){
                sell[j] = max(sell[j],buy[j]+prices[i]);
                if(j==0){
                    buy[j] = max(buy[j],-prices[i]);
                }else{
                    buy[j] = max(buy[j],sell[j-1]-prices[i]);
                }
            }
        }
        return sell[k-1];
    }
};