[动态规划] 股票买卖

描述

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

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

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

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

输入

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

输出

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

样例输入

3
7
5 14 -2 4 9 3 17
6
6 8 7 4 1 -2
4
18 9 5 2

样例输出

28
2
0

解题分析

首先考虑到较为巨大的数据读入量，我们先写一个快速读入。首先是对空格、换行符等符号进行getchar()读入，遇到数字或者负号 “-‘’的时候停止，然后判断一下符号，是正还是负。之后利用while去不断读取字符，并利用字符转数字的神秘数字48进行更新，最后返回读入的数字。

接着，我们思考一下本题应该如何去做，实际上，很容易想到去枚举全部的情况。不过这样的话整个题目的运算量就大起来了，那怎么利用动态规划的思想去减少我们的计算量呢？这里就要考虑分割的线性规划的思想了。我们可以定义两个数组pre和after，pre[i]表示到第i天我们可以买卖一次股票得到的最大利润，after[i]表示从第i天往后我们可以买卖一次股票得到的最大利润，最后我们再遍历一遍结果取最大值即可。

代码实现

#include <cstdio>
#include <climits>
#include <cctype>
#include <algorithm>
using namespace std;

inline int in(){
	char x=getchar(); int f=1,go=0;
	while (!isdigit(x) && x!='-') x=getchar();
	if (x=='-') f=-f,x=getchar();
	while (isdigit(x)) go=go*10+x-48,x=getchar();
	return go*f;
} 

int stock[100005]={0};
int pre[100005]={0};
int after[100005]={0};

int main(){
	int T;
	T=in();
	while(T--){
		int N=in();
		for(int i=1;i<=N;i++){
			stock[i]=in();
		}
		int minn=1e9;
		pre[0]=after[N+1]=0;
		for(int i=1;i<=N;i++){
			minn=min(minn,stock[i]);
			pre[i]=max(pre[i-1],stock[i]-minn);
		}
		int maxn=-1e9;
		int ans=-1e9;
		for(int i=N;i>=1;i--){
			maxn=max(maxn,stock[i]);
			after[i]=max(after[i+1],maxn-stock[i]);
			ans=max(pre[i]+after[i],ans);
		}
		printf("%d\n",ans);
	}
	return 0;
}