【小米】风口的猪-中国牛市

题目描述

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

示例1

输入

3,8,5,1,7,8

输出

12

分析：

因为要求两次交易后的最大收益，所以先正序扫一遍记录第k天能获得的最大收益，然后逆序扫表示第k天之后的区间能获得的最大收益，然后相加求最大值

利用动态规划的思想。

首先从左往右扫描，先计算出子序列[0,...,i]中的最大利润，用一个left数组保存下来，时间是O(n)。第二步是从右往左逆向扫描，计算子序列[i,...,n-1]上的最大利润，

用right数组保存下来，s时间是O(n)，同时结合left和right数组的结果就能计算最终的最大利润了，这一步也是O(n)。 所以最后算法的复杂度就是O(n)的。

code:

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