Best Time to Buy and Sell Stock III

Best Time to Buy and Sell Stock III

 Total Accepted: 30820 Total Submissions: 130535My Submissions

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

#include <vector>
#include <iostream>
using namespace std;


int main()
{
    int priSize ,tem,i;
    vector<int> pri;
    while(cin >> priSize)
    {
        pri.clear() ;
        for( i =  0; i < priSize ; ++ i)
        {
            cin >> tem ;
            pri.push_back(tem);
        }

        if(priSize<=1) 
        {
            cout << 0 << endl;
            continue;
        }
        if(priSize==2)  
        {
            int tem = pri[1]>pri[0]?pri[1]-pri[0]:0;
            cout << tem << endl;
            continue;
        }


        //第一种情况，只买了一次
        // 1、从左到右遍历 保存最小价格MIN ，判断 pri[i] - MIN > MaxPrice ? ,更新 当前最大利润 MaxPrice
        // 2、从左到右遍历 保存最大价格MAX ，判断 MAX - pri[i] > MaxPrice ? ,更新 MaxPrice

        //第二种情况，买了两次
        //3、max( d ) = MaxPrice( 0 <= d <= i ) + MaxPrice( i < d < priSize ) 
        //d  在 0 < d < priSize - 1 之间遍历 ，判读 max( d ) > MaxPrice , 更新 MaxPrice
        //MaxPrice( 0 <= d <= i ) 和   MaxPrice( i < d < priSize )   的结果
        //可以在 第 1、2 步 在 maxFromLeft，maxFromRight 预先记录下来
        //4、maxFromLeft[priSize-1] == maxFromRight[0] == 总共只买一次的最大利润
        //用它们之中的一个来 和 第 3、中的 MaxPrice 比较大小，得出结果


        //算法复杂度 O( n-1 +  n-1 + n - 2 ) = O(3n - 4) = O( n ) 

        vector<int> maxFromLeft(priSize,0);    
        vector<int> maxFromRight(priSize,0);
        int MIN = INT_MAX, MAX=INT_MIN, MaxPrice = INT_MIN,priTem;
        for(int i=0;i<priSize;i++)
        {
            //更新从左边开始的最小价格
            if(pri[i]<MIN)
                MIN = pri[i];
            priTem = pri[i]- MIN;

            //更新当前最大盈利
            if(priTem > MaxPrice)
                MaxPrice = priTem;

            //保存 MaxPrice( 0 <= d <= i )
            maxFromLeft[i] = MaxPrice;
        }

        MaxPrice = INT_MIN ;
        for(int i=priSize-1;i>=0;--i)
        {
            //更新从右边开始的最大价格
            if(pri[i]>MAX)
                MAX = pri[i];
            priTem = MAX - pri[i];
            //更新当前最大盈利
            if(priTem > MaxPrice)
                MaxPrice = priTem;
            //保存  MaxPrice( i < d < priSize )
            maxFromRight[i] = MaxPrice;
        }

        MaxPrice = INT_MIN ;
        for(int i=0;i<priSize-1;i++){
            //更新当前最大盈利
            priTem = maxFromLeft[i]+maxFromRight[i+1];
            if(priTem > MaxPrice) MaxPrice = priTem;
        }

        //maxFromLeft[priSize-1] 和  MaxPrice 比较大小，得出结果
        if( maxFromLeft[priSize-1] > MaxPrice )
            MaxPrice = maxFromRight[0];

        cout << MaxPrice << endl;
    }

    return 0 ;
}