123.Best Time to Buy and Sell Stock III(H)

题目描述

给定一个数组prices，数组中的每个元素分别代表第i天股票的价格。现在你能进行最多两次交易，求你能取得的最大利益。
备注：同时只能持有一支股票。
原题如下，

算法描述

这道题如果要穷举出所有的情况，肯定是不现实的。我们用这么一个贪心的策略来得到我们想要的结果。
我们记录4个状态，分别是第一次购买股票时获取的利益（firstBuy），第一次卖出股票时获取的利益（firstSell），第二次购买股票时获取的利益（secondBuy），第二次出售股票时获取的利益（secondSell）。我们这样来考虑，如果要使总的利益最大，那么第二次出售时便应获取最大利益；要使第二次出售时获取最大利益，那么就要使第二次购买时获取最大利益，一次类推，我们可以得到以下的公式，
firstBuy = max(firstBuy, -prices[i])
firstSell = max(firstBuy + prices[i], firstSell)
secondBuy = max(firstSell - prices[i], secondBuy)
secondSell = max(secondBuy + prices[i], secondSell)

源代码如下，

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<int>  dp(4, 0);
        dp[0] = INT_MIN;
        dp[2] = INT_MIN;

        for (int i = 0; i < prices.size(); ++i) {
            dp[0] = max(-prices[i], dp[0]);
            dp[1] = max(dp[1], dp[0] + prices[i]);
            dp[2] = max(dp[2], dp[1] - prices[i]);
            dp[3] = max(dp[3], dp[2] + prices[i]);
        }

        return dp[0][3];
    }
};