LeetCode OJ 123 Best Time to Buy and Sell Stock III

最新推荐文章于 2019-05-01 17:57:00 发布

原创最新推荐文章于 2019-05-01 17:57:00 发布 · 542 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode OJ刷题日记专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一种算法，用于确定在给定的股票价格序列中如何通过最多两次交易获得最大利润。该算法首先从前向后扫描，计算每个位置左侧可能获得的最大利润；然后从后向前扫描，计算每个位置右侧可能获得的最大利润。最终通过合并这两个阶段的结果找到最大利润。

题目：

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

难度：hard

思路：正向扫面在i处左边可得的最大利润，逆向扫面在i处右边可得的最大利润，再将所得的两个数组相加求最大值

代码如下

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<int> profit1(prices.size(),0);
        vector<int> profit2(prices.size(),0);
        if(prices.size()<=1)
            return 0;
        int minPrice=prices[0];
        for(int i=1;i<prices.size();i++){
            minPrice= min(minPrice,prices[i]);
            profit1[i]=max(prices[i]-minPrice,profit1[i-1]);
        }
        
        int maxPrice=prices[prices.size()-1];
        for(int i=prices.size()-2;i>=0;i--){
            maxPrice = max(maxPrice,prices[i]);
            profit2[i]=max(maxPrice-prices[i],profit2[i+1]);

        }
        int maxProfit=0;
        for(int i=0;i<prices.size();i++){
            maxProfit=max(maxProfit,profit1[i]+profit2[i]);
        }
        return maxProfit;
    }
};