leetcode——188——Best Time to Buy and Sell Stock IV

Say you have an array for which the ith element is the price of a given stock on day
i.

Design an algorithm to find the maximum profit. You may complete at most
k transactions.

Note:

You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).

此题与那个有相似之处，也有不同。这个题要求收益最大，也即在一个数组中找顺序数字的最大差值。当然此题还没那么简单，还要求可以最多交易k次。举例来说，如果给定一个数组是1,3,7,2,1,5 ，那么很明显最小值是1（注意虽然有两个数字1，但是第二个1是不可取的，因为我们为了获取最大收益要在7处卖掉，所以在那之前必须买入），最大值是7，所以只交易一次的最大收益即为6，同理交易两次最大收益是10（这应该很容易看出来吧？）

类似最大连续和问题，我们需要变量来记录当前的一些状态信息，因为是k次交易，所以我们需要大小为k的数组记录每一次交易的信息，又因为一次交易包含买和卖两次操作，所以我们需要两个数组。分别命名为hold【】和sell【】。hold[i]表示第i次持有后的盈亏，也就是第i次买入后的状态，sell[i]表示第i次卖出后的盈亏状态。比如上个例子1,3,7,2,1,5.如果我们在第一天买入，那么hold[1]应该是-1，这里可能会问，为什么会是负数？那是因为我们买入了股票，钱已经花出去了，但我们还没有卖出，所以当前你手上是没有现金的，只有股票。

算法核心就是两句：

cur = prices[i]; hold[j] =
max(hold[j],sell[j-1] - cur); sell[j] =
max(sell[j],hold[j] + cur);

在第i天第j次持有的盈亏最大是取两个的较大值，前一个表示继续持有，后一个表示在第j-1次卖出后，第i天又买入了。

在第i天第j次卖出的盈亏最大同上，也是比较两数取较大。

class Solution {
public:
int maxProfit(int k, vector<int>& prices) {
int len = prices.size();
if (len<2) return 0;
if (k>len/2)
{ // simple case
int ans = 0;
for (int i=1; i<len; ++i)
{
ans += max(prices[i] - prices[i-1],0);
}
return ans;
}
int hold[k+1];
int sell[k+1];
for (int i=0;i<=k;++i)
{
hold[i] = INT_MIN;
sell[i] = 0;
}
int cur;
for (int i=0; i<len; ++i)
{
cur = prices[i];
for(int j=1; j<=k; ++j)
{
sell[j] = max(sell[j],hold[j] + cur);
hold[j] = max(hold[j],sell[j-1] - cur);
}

}
return sell[k];

}
};