买卖股票的最佳时机 IV -LintCode

假设你有一个数组，它的第i个元素是一支给定的股票在第i天的价格。

设计一个算法来找到最大的利润。你最多可以完成 k 笔交易。

注意事项：

你不可以同时参与多笔交易(你必须在再次购买前出售掉之前的股票)

样例：

给定价格 = [4,4,6,1,1,4,2,5], 且 k = 2, 返回 6.

挑战 ：

O(nk) 时间序列。

思路见： http://blog.csdn.net/linhuanmars/article/details/23236995

#ifndef C393_H
#define C393_H
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
/*
* @param K: An integer
* @param prices: An integer array
* @return: Maximum profit
*/
int maxProfit(int K, vector<int> &prices) {
// write your code here
if (prices.empty())
return 0;
int len = prices.size();
if (K > len / 2)
{
int max = 0;
for (int i = 1; i < len; ++i)
{
int diff = prices[i] - prices[i - 1];
max += maxVal(0, diff);
}
return max;
}
vector<int> global(K + 1, 0);
vector<int> local(K + 1, 0);
for (int i = 0; i < len - 1; ++i)
{
int diff = prices[i + 1] - prices[i];
for (int j = K; j >= 1; --j)
{
local[j] = maxVal(global[j - 1] + maxVal(diff, 0), local[j] + diff);
global[j] = maxVal(global[j], local[j]);
}
}
return global[K];
}
int maxVal(int a, int b)
{
return a > b ? a : b;
}
};
#endif