动态规划练习：股票买卖 线性dp

题目描述

最近越来越多的人都投身股市，阿福也有点心动了。谨记着“股市有风险，入市需谨慎”，阿福决定先来研究一下简化版的股票买卖问题。

 

假设阿福已经准确预测出了某只股票在未来N天的价格，他希望买卖两次，使得获得的利润最高。为了计算简单起见，利润的计算方式为卖出的价格减去买入的价格。

同一天可以进行多次买卖。但是在第一次买入之后，必须要先卖出，然后才可以第二次买入。

 

现在，阿福想知道他最多可以获得多少利润。

 

输入

输入的第一行是一个整数T(T≤50)，表示一共有T组数据。

接下来的每组数据，第一行是一个整数N(1≤N≤100,000)，表示一共有N天。第二行是 N 个被空格分开的整数，表示每天该股票的价格。该股票每天的价格的绝对值均不会超过1,000,000。

 

输出

对于每组数据，输出一行。该行包含一个整数，表示阿福能够获得的最大的利润。

  

样例输入1
3
7
5 14 -2 4 9 3 17
6
6 8 7 4 1 -2
4
18 9 5 2

样例输出1
28
2
0

 

 

思想：

这种题一看就是dp，没什么可说的

看题，他说如果已经有股票了必须先卖后买

我们可以把买股票到卖股票抽象成一个序列，因为共买卖两次，所以把整个序列分成两部分,并且必须是不相交的两部分

 

动态规划三要素，不明白的可以点这里

 

状态定义：定义两个数组，l和r

l[i]表示在第i天卖掉股票的最大利润

r[i]表示在第i天买股票的最大利润

状态初始化：全部为0

状态转移方程：l[i]=max(l[i-1],a[i]-buy1);//其中buy1为第i天之前价格的最低

r[i]=max(r[i-1],buy2-a[i]);//buy2表示第i天及以后价格的顶峰

# include <iostream>
# include <cstdio>
# include <cstring>
using namespace std;
# define int long long
# define N 100005
int n,l[N],r[N];//第i天前最大利润和第i天后的最大利润 
int t,sum;
int a[N];
signed main(){
	scanf("%lld",&t);
	while(t--){
		memset(l,0,sizeof(l));
		memset(r,0,sizeof(r));
		sum=0;
		scanf("%lld",&n);
		for (int i=1;i<=n;i++){
			scanf("%lld",&a[i]);
		}
		int buy1=a[1];
		int buy2=a[n];
		for (int i=1;i<=n;i++){
			l[i]=max(l[i-1],a[i]-buy1);
			buy1=min(a[i],buy1);
		}
		for (int i=n;i>=1;i--){
			r[i]=max(r[i+1],buy2-a[i]);
			buy2=max(a[i],buy2);
		}
		for (int i=1;i<=n;i++){
			sum=max(sum,l[i]+r[i]);
		}
		if (sum<0){//如果赔了还不如不买 详见样例第3组数据
			printf("0\n");
			continue;
		}
		printf("%lld\n",sum);
	}
	return 0;
}