给定一个长度为 N 的数组,数组中的第 i 个数字表示一个给定股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成两笔交易。
注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
输入格式
第一行包含整数 N,表示数组长度。
第二行包含 N 个不大于 109 的正整数,表示完整的数组。
输出格式
输出一个整数,表示最大利润。
数据范围
1≤N≤105
输入样例1:
8
3 3 5 0 0 3 1 4
输出样例1:
6
输入样例2:
5
1 2 3 4 5
输出样例2:
4
输入样例3:
5
7 6 4 3 1
输出样例3:
0
样例解释
样例1:在第 4 天(股票价格 = 0)的时候买入,在第 6 天(股票价格 = 3)的时候卖出,这笔交易所能获得利润 = 3-0 = 3 。随后,在第 7 天(股票价格 = 1)的时候买入,在第 8 天 (股票价格 = 4)的时候卖出,这笔交易所能获得利润 = 4-1 = 3 。共得利润 3+3 = 6。
样例2:在第 1 天(股票价格 = 1)的时候买入,在第 5 天 (股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。注意你不能在第 1 天和第 2 天接连购买股票,之后再将它们卖出。因为这样属于同时参与了多笔交易,你必须在再次购买前出售掉之前的股票。
样例3:在这种情况下, 不进行任何交易, 所以最大利润为 0。
代码
#include <iostream>
using namespace std;
int n,m,minv,maxv,res;//全局变量默认初始化为0
int p[100005],f[100005],s[100005];
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
cin>>p[i];//输入这n天股票的价格
//f存储着在第i天卖出的最大收益,在股票价最低时买进最高时卖出能获得最大收益
//f[i]=max(第i天不卖出,第i天的股票价-前i-1天中的最低股票价)
//第1天买进,然后卖出总收益为0(因为买卖价格相同),
//数组本来就初始化为0,所以可以从2开始。初始化的最低股票价是第1天的股票价
for(int i=2,minv=p[1];i<=n;i++){
f[i]=max(f[i-1],p[i]-minv);
minv=min(minv,p[i]);//更新前i天的最低股票价
}
//s存储着在第i天买进的最大收益,在股票价最低时买进最高时卖出能获得最大收益
//s[i]=max(第i天不买进,i+1天之后的最高股票价-第i天的股票价)
//初始化最高股票价是第n天的股票价
for(int i=n-1,maxv=p[n];i>=1;i--){
s[i]=max(s[i+1],maxv-p[i]);
maxv=max(maxv,p[i]);//更新第i天之后的最高股票价
}
//一个问题分为两段的话,可以先考虑一段的情况
//再划分所有可能的分段点,遍历所有分段的可能
//注意:因为最多可以完成两笔交易,那么你也可以只进行一笔交易
//因为当i=1的时候代表第一天买进又卖出。其收益为0
//即相当于只进行在2-n天中买进卖出的一笔交易
//当i=n时只进行在1-n天中买进卖出的一笔交易包含i=1的情况,所以i也可以从2开始
for(int i=1;i<=n;i++)//res默认初始化为0
res=max(res,f[i]+s[i+1]);
cout<<res<<endl;
return 0;
}