第十周 Best Time to Buy and Sell Stock III

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