题目:
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 two transactions.
Note:
You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).
class Solution {
public:
//DP
//最多两笔交易能取得的最大利润
int maxProfit(vector<int>& prices) {
if (prices.empty()) return 0;
int *tbl1 = new int[prices.size()];//tbl1[i]表示第i天可以取得的最大利润
int *tbl2 = new int[prices.size()];//tbl2[i]表示从第i天到最后一天可以取得的最大利润
int min = prices[0];//min表示i之前的最小值
int max = prices[prices.size() - 1];//max表i之后的最大值
tbl1[0] = 0;
tbl2[prices.size() - 1] = 0;
//如何在第0...i天获得最大利润?肯定是历史最低价买进,那什么时候卖出呢?
//从i=0开始每天有两种选择:卖或不卖
//当天卖能取得最大利润是:当天价格减去历史最低价
//当天不卖能取得最大利润是:前一天能取得的最大利润
//当天能取得的最大利润是以上两者中的最大值
for (int i = 1; i < prices.size(); ++i){
if (prices[i] < min) min = prices[i];//如果当天价格比历史最低价低,则更新历史最低价
if (prices[i] - min > tbl1[i - 1]) tbl1[i] = prices[i] - min;//如果当天价格减去历史最低价(当天卖出)取得的利润比前一天取得最大利润大,则当天卖出
else tbl1[i] = tbl1[i - 1];//否则当天不卖出(在当天之前卖出)
}
//如何在第i...prices.size()-1天获得最大利润?肯定是在i...prices.size()-1中的最高价卖出,那什么时候买进呢?
//从最后一天开始,每天有两种选择:买或不买
//当天买能取得最大利润是:未来的最高价减去当天价格
//当天不买能取得最大利润是:后一天能取得的最大利润
//当天到最后一天能取得的最大利润是以上两者中的最大值
for (int i = prices.size() - 2; i >= 0; --i){
if (prices[i] > max) max = prices[i];
if (max - prices[i] > tbl2[i + 1]) tbl2[i] = max - prices[i];//当天买进
else tbl2[i] = tbl2[i + 1];//当天不买(以后再买)
}
int res = tbl1[0] + tbl2[0];
for (int i = 1; i < prices.size(); ++i){
if (tbl1[i] + tbl2[i] > res) res = tbl1[i] + tbl2[i];
}
delete[]tbl1;
delete[]tbl2;
return res;
}
};