codeforces #331 D. Wilbur and Trees (dp)

题目：http://codeforces.com/contest/596/problem/D

题意：有n棵高为h的树在一排，已知每棵树的位置pos。现在你从最左边或者从最右边等概率挑选一棵树砍断，树向左倒的概率为P向右倒的概率为（1-P），树倒下时如果撞到了其它的树会发生连锁反应，假如还有树没有倒完，你还会等概率挑选一棵树砍断。现在问你所有树倒下时覆盖地面的长度的期望。

分析：

开始定义dp[i][j]为区间[i,j]的答案，然后枚举两端的树倒下的方向，发现推不出，因为当枚举第i棵树向左倒的时候不知道是否会与第i-1棵树发生重叠，所以还需要知道第i-1棵树倒下的方向。

重新定义dp数组，定义dp[i][j][f1][f2]为第i-1棵树倒下的方向为f1，第j+1棵树倒下的方向为f2时区间[i,j]的解。

由于会发生连锁反应，所以要先预处理出每一棵树向哪个方向倒下时会影响的最远的树的标号。

那么我们枚举四种情况：①选树i，树i向左边倒②选树j，树j向右边倒③选树i，树i向右边倒④选树j，树j向左边倒

PS：当由区间[x,y]递推到区间[x',y]时，说明x倒下不会使x'也倒下。

代码参考：点击打开链接

代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
typedef unsigned long long ULL;
const LL INF = 1E9+9;
const int maxn = 2006;
int pos[maxn];
double dp[maxn][maxn][2][2];
bool visit[maxn][maxn][2][2];
int h,n,dl[maxn],dr[maxn];
double P;

double dfs(int l,int r,int f1,int f2)
{
	if(visit[l][r][f1][f2])
		return dp[l][r][f1][f2];
	if(l>r)
		return 0;
	visit[l][r][f1][f2]=1;
	double &ans=dp[l][r][f1][f2];

	ans+=0.5*P*(min(pos[l]-pos[l-1]-h*f1,h)+dfs(l+1,r,0,f2));
	ans+=0.5*(1-P)*(min(pos[r+1]-pos[r]-h*f2,h)+dfs(l,r-1,f1,0));
		
	int L=dr[l];
	int R=dl[r];
	if(L<r)
		ans+=0.5*(1-P)*(pos[L]-pos[l]+h+dfs(L+1,r,1,f2));
	else
		ans+=0.5*(1-P)*(pos[r]-pos[l]+min(pos[r+1]-pos[r]-h*f2,h));

	if(R>l)
		ans+=0.5*P*(pos[r]-pos[R]+h+dfs(l,R-1,f1,1));
	else
		ans+=0.5*P*(pos[r]-pos[l]+min(pos[l]-pos[l-1]-h*f1,h));

//	printf("%d %d %d %d  %.15lf\n",l,r,f1,f2,ans);
	return ans;
}

int main()
{
	int i,j,k;
	scanf("%d%d%lf",&n,&h,&P);
	for(i=1;i<=n;i++)
		scanf("%d",&pos[i]);
	sort(pos+1,pos+n+1);
	pos[0]=-INF;
	pos[n+1]=INF;
	dr[n]=n;
	for(i=n-1;i>=1;i--)
		dr[i]=(h>pos[i+1]-pos[i]?dr[i+1]:i);
	dl[1]=1;
	for(i=2;i<=n;i++)
		dl[i]=(h>pos[i]-pos[i-1]?dl[i-1]:i);
	printf("%.10lf\n",dfs(1,n,0,0));
	return 0;
}