Best Time to Buy and Sell Stock III
LeetCode上的123题:https://leetcode.com/problems/best-time-to-buy-and-sell-stock-iii/
题目
给出一个大小为n的数组,表示从1到n天股票的价格。有两次交易机会(一次交易包括买和卖),一次交易完成后才能进行下一次交易,求能得到的最大利润。
Example1:
Input: [3,3,5,0,0,3,1,4]
Output: 6
Explanation: Buy on day 4 (price = 0) and sell on day 6 (price = 3), profit = 3-0 = 3.
Then buy on day 7 (price = 1) and sell on day 8 (price = 4), profit = 4-1 = 3.
Example2:
Input: [1,2,3,4,5]
Output: 4
Explanation: Buy on day 1 (price = 1) and sell on day 5 (price = 5), profit = 5-1 = 4.
Note that you cannot buy on day 1, buy on day 2 and sell them later, as you are
engaging multiple transactions at the same time. You must sell before buying again.
Example3:
Input: [7,6,4,3,1]
Output: 0
Explanation: In this case, no transaction is done, i.e. max profit = 0.
题解
我们令p[i][j]表示在第j天,进行了i次交易后得到的最大利润,那么问题就变成求p[2][n]。有人会问,在例子2中只进行了1次交易就得到了最大利润,求p[2][n]是不是不正确。其实例2还可以在第五天卖出股票后再以相同的价格买卖股票,这样第二次交易利润为0,两次交易同样得到最大利润。所以p[i][n]一定大于等于p[i-1][n],有交易一次得到最大利润的方法就一定有交易两次的方法。我们可以得出,在同一天买入和卖出只会减少一次交易机会,是没有意义的,所以我们一般不会考虑同一天买和卖。
为了将p[i][j]转成小问题,考虑第j天是否卖出股票,如果不卖股票,p[i][j] = p[i][j-1], 如果卖股票,p[i][j] = prices[j] - min_cost, 其中min_cost表示第i-1次交易的前j天股票的最小成本。
这么说可能有些拗口。我们先来看看什么是股票的(最低)成本。对于数组[1,2,3,4,5],在第1次交易,每天的股票成本显然是7,但在第二次交易,第三天的股票成本是2,因为在前两天可以发生一次交易得到最大利润2-1。因此股票的成本等于cost[i][j] = prices[j]-p[i-1][j-1];
而min_cost就表示前j天股票成本的最小值。题目的基准情况:p[0][i] = 0 没有交易机会就没有利润。
p[k][1]只有一天的股票交易机会也没有利润。
代码
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>> profit(3, vector<int>(prices.size(), 0)); //初始化二维vector满足基准情况
for (int i = 1; i < 3; i++) { //便利每次交易
int cost = prices[0]; //无论是第几次交易,第一天的股票成本一定是股票价格
for (int j = 1; j < prices.size(); j++) {
cost = min(cost, prices[j] - profit[i-1][j-1]); //找到第i-1次交易的前j天的最小股票成本
profit[i][j] = max(profit[i][j-1], prices[j]-cost); //max的第一个参数是不交易情况得到的利润,第二次是交易得到的最大利润
}
}
return profit[2][prices.size()-1];
}
};