风口的猪-中国牛市

题目描述

风口之下，猪都能飞。当今中国股市牛市，真可谓“错过等七年”。 给你一个回顾历史的机会，已知一支股票连续n天的价格走势，以长度为n的整数数组表示，数组中第i个元素（prices[i]）代表该股票第i天的股价。 假设你一开始没有股票，但有至多两次买入1股而后卖出1股的机会，并且买入前一定要先保证手上没有股票。若两次交易机会都放弃，收益为0。 设计算法，计算你能获得的最大收益。 输入数值范围：2<=n<=100,0<=prices[i]<=100 

IDEA

总体来说，是求两次交易的最大收益。

正序遍历：求第i天时第一次交易获得的最大收益

倒序遍历：求第i天时，后面有第二次交易可获得的最大收益

然后将第i天的两次最大收益相加，比较，得出最终的最大收益

CODE

class Solution {
public:
    /**
     * 计算你能获得的最大收益
     * 
     * @param prices Prices[i]即第i天的股价
     * @return 整型
     */
    int calculateMax(vector<int> prices) {
        int n=prices.size();
        vector<int> leftmax(n);
        vector<int> rightmax(n);
        
        int lmin=prices[0];
        leftmax[0]=0;
        for(int i=1;i<n;i++){
            lmin=min(lmin,prices[i]);
            leftmax[i]=max(leftmax[i-1],prices[i]-lmin);
        }
        
        int rmax=prices[n-1];
        leftmax[n-1]=0;
        for(int i=n-2;i>=0;i--){
            rmax=max(rmax,prices[i]);
            rightmax[i]=max(rightmax[i+1],rmax-prices[i]);
        }
        
        int sum=0;
        for(int i=0;i<n;i++){
            sum=max(sum,leftmax[i]+rightmax[i]);
        }
        return sum;
    }
};

简短：

public class Solution {
    /**
     * 计算你能获得的最大收益
     * 
     * @param prices Prices[i]即第i天的股价
     * @return 整型
     */
    public int calculateMax(int[] prices) {
        int firstBuy=Integer.MAX_VALUE;
        int firstSell=0;
        int secondBuy=Integer.MAX_VALUE;
        int secondSell=0;
 
        for (int curPrice:prices) {
            firstBuy=Math.min(firstBuy,curPrice);//用最低的价钱买
            firstSell=Math.max(firstSell,curPrice-firstBuy);//收益最高的时候第一次卖出
            secondBuy=Math.min(secondBuy,curPrice-firstSell);//用第一次赚的钱进行第二次买，所以要减去firstSell
            secondSell=Math.max(secondSell,curPrice-secondBuy);//收益最高的时候第二次卖出
        }
        return secondSell;
    }
}