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];
}
};