TZOJ 8501: 股票卖卖最佳时机3

描述

给定一支股票若干天的价格，设计一个算法来计算你所能获取的最大利润。你最多可以完成两笔交易。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

输入

第一行为n（1≤n≤105），表示天数。

第二行为n个整数ai (0≤ai≤104)，ai表示第i天的股票价格。

输出

输出最大利润，如果你不能获取任何利润，输出0。

样例输入

7
2 1 4 5 2 9 7

样例输出

11
 

题解

本题股票最多可以买卖2次

dp数组含义 
dp[i][0]：到第i天为止不操作 
dp[i][1]：到第i天为止第一次持有
dp[i][2]：到第i天为止第一次不持有
dp[i][3]：到第i天为止第二次持有
dp[i][4]：到第i天为止第二次不持有

dp数组初始化 
dp[0][1]=-w[0]：到为止第1天持有的最大收益为-w[0] 
dp[0][3]=-w[0]：到为止第1天持有的最大收益为-w[0] 

dp数组递推公式：
dp[i][1]=max(dp[i-1][1],-w[i]);
dp[i][2]=max(dp[i-1][2],dp[i-1][1]+w[i]);
dp[i][3]=max(dp[i-1][3],dp[i-1][2]-w[i]);第二次持有可能从第一次不持有的情况来 
dp[i][4]=max(dp[i-1][4],dp[i-1][3]+w[i]);
本天持有的状态只有可能从前一天持有的状态和前一天不持有本天买入来，不持有的状态一样

代码

#include<bits/stdc++.h>
using namespace std;
const int N=100005;
int dp[N][5],w[N];
//dp数组含义 
//dp[i][0]：到第i天为止不操作 
//dp[i][1]：到第i天为止第一次持有
//dp[i][2]：到第i天为止第一次不持有
//dp[i][3]：到第i天为止第二次持有
//dp[i][4]：到第i天为止第二次不持有
int main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int n;
	cin>>n;
	for(int i=0;i<n;i++)cin>>w[i];
	dp[0][0]=0;
	dp[0][1]=-w[0];//到为止第1天持有的最大收益为-w[0]
	dp[0][2]=0;
	dp[0][3]=-w[0];//到为止第1天持有的最大收益为-w[0]
	dp[0][4]=0;
	for(int i=1;i<n;i++){
		//dp数组递推公式：
		dp[i][1]=max(dp[i-1][1],-w[i]);
		dp[i][2]=max(dp[i-1][2],dp[i-1][1]+w[i]);
		dp[i][3]=max(dp[i-1][3],dp[i-1][2]-w[i]);//第二次持有可能从第一次不持有的情况来 
		dp[i][4]=max(dp[i-1][4],dp[i-1][3]+w[i]);
		//本天持有的状态只有可能从前一天持有的状态和前一天不持有本天买入来，不持有的状态一样
	}
	cout<<dp[n-1][4]<<endl;
}