Best Time to Buy and Sell Stock III(Leetcode)

Best Time to Buy and Sell Stock III(Leetcode)

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).

解题思路：

动态规划。用二位数组 v[j][i] 表示考虑到第 i 个数的第 j 次 transaction 的最大值。

考虑如何得到 v[j][i] 。对于第 i 个数有两种情况：

①卖出的第 i 天的股票。这就需要在第 i 天之前要买进股票，所以从第 i-1 天开始往前面找。对于历遍的数又有两种情况：

a. 该数值小于第 i 天股票的值。所以可以得到 v[j][i] = max(v[j][i], v[j-1][t-1] + prices[i] - prices[t]) 。（t是往回找的数的下标）

b. 该数值大于等于第 j 天股票的值。这种情况下可以得到 v[j][i] = max(v[j][i], v[j][t])。然后直接break跳出往回找的循环就可以了。

    因为该数值（第 t 天的值）大于等于第 i 天的值，也就是说第 t 天和第 i 天不可能构成一次transaction，而第 t 天之后的数值在往回找的过程中已经历遍了，所以此时第 i 天只能和第 t 天之前的某一天构成一次transaction。但是因为 prices[t] > prices[j] ，所以此时 v[j][t] > v[j][i] 。

②不卖出也不买入第 j 天的股票。v[j][i] = v[j][i-1] 。

代码：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        return maxProfit(prices, 2);
    }
    int maxProfit(vector<int>& prices, int n) {
        if (prices.size() == 0) return 0;
        vector<vector<int>> v(n, vector<int>(prices.size(), 0));
        for (int i = 1; i < prices.size(); ++i) {
            for (int j = 0; j < n; ++j) {
                v[j][i] = v[j][i-1];
                for (int t = i-1; t >= 0; --t) {
                    if (prices[t] < prices[i]) {
                        if (j == 0 || t == 0) {
                            v[j][i] = max(v[j][i], prices[i] - prices[t]);
                        } else {
                            v[j][i] = max(v[j][i], prices[i] - prices[t] + v[j-1][t-1]);
                        }
                    } else {
                        v[j][i] = max(v[j][i], v[j][t]);
                        break;
                    }
                }
            }
        }
        return v[n-1][prices.size()-1];
    }
};