小米笔试(风口的猪-中国牛市)

xiaoxiao2021-02-28  109

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

输入

3,8,5,1,7,8

输出

12    这是一道动态规划的题

动态规划思路:定义数组left[n],left[i]表示股票在当天能获得的最大收益,状态转移方程则可以表示为:left[i]=prices[i]-min>prices[i]?prices[i]-min:prices[i-1].

对于第二次买入卖出定义数组right[n],同理right[i]表示当天能获得最大收益。不过left数组是算出当前为止的最低价格并算出当天为止的最大差价。right数组是倒序算出最高价格并记录最大差价

max是倒序到当天的最高值,过程同上

总结一下,这道题有四个动态变化量,分别为第一次买入卖出的最小值min和最大差价left[i],min为直到当天股市的最低值,初始化为prices[0],最大利润left[i]初始化为0。如果当天卖出,(不考虑当天买当天卖的情况,遍历从第二天开始),则收益为当天价格prices[i]-min,如果大于到目前为止最大利润left[i-1],则更新left[i-1];否则最大利润继续为left[i-1],和前一天保持一致;

另外两个动态变化量为max和right[i],同理

至此我们就可以写出如下代码,采用两个数组双向遍历,避免重叠问题

class Solution { public: /** */ //以卖股票当日为分界点,另外注意股票可以当天卖了再买 int calculateMax(vector<int> prices) { int n=prices.size(); //下两个变量代表最大的差价 vector<int>left(n,0);//第一次买入卖出最大差价 vector<int>right(n,0);//第二次买入卖出最大差价。为了防止两次重叠问题,第二次从右向左遍历 int max=0,min=0; min=prices[0]; max=prices[n-1]; for(int i=1;i<n;i++)//从第二天就算与第一天的差价开始 { if(prices[i]<min) min=prices[i]; left[i]=prices[i]-min>left[i-1]? prices[i]-min:left[i-1];//当前的价格和目前为止最低价相减与上一日为止最大差价比较 } for(int j=n-2;j>=0;j--) { if(prices[j]>max) max=prices[j]; right[j]=max-prices[j]>right[j+1]?max-prices[j]:right[j+1]; } max=0; for(int i=0;i<n;i++) if(left[i]+right[i]>max) max=left[i]+right[i]; return max; } };

恭喜通过本题

恭喜你通过本题

运行时间:3ms

占用内存:496k

同步到: 进入下一题 分享解题思路

恭喜通过本题

恭喜你通过本题

运行时间:3ms

占用内存:496k

同步到: 进入下一题 分享解题思路
转载请注明原文地址: https://www.6miu.com/read-52798.html

最新回复(0)