CF596D Wilbur and Trees

一、题目

点此看题

$luogu$讨论里面有翻译哟，自己去看呗

二、解法

只会从两个端点砍树，很容易看出来是区间$dp$了，但是一个区间$[l,r]$的$l$向左倒还是向右倒会影响到$[l+1,r]$的期望，但是问题不大，我们可以直接把端点的情况塞进状态里面，于是就有了状态定义：设$f[l][r][x][y]$为砍$[l,r]$这个区间里的树，第$l-1$棵树的倾倒情况为$x$，第$r+1$棵树的倾倒情况为$y$（其中$0/1$表示向外$/$内倾倒），所得到的区域覆盖期望。

出口就是$l>r$的情况，转移肯定是讨论四周情况，我就举左端点的例子（右端点类推），第一种是左端点向左倒，$l$的区域覆盖贡献即为$\min (a[l]-a[l-1]-xh,h)$，然后我们再去找$f[l+1][r][0][y]$（注意由于本题只要距离小于$h$就会连着倒，所以上述情况$a[l]-a[l-1]$一定是不小于$h$的），第二种就是左端点向右倒，处理出连着倒最多到达的点$cr[l]$（$O(n)$很容易就能预处理），分为一片直接倒完和没有倒完的情况转移，至于区域覆盖贡献的计算画个图就能理解了。

时间复杂度$O(n^2)$，代码还是很简洁的$qwq$

#include <cstdio>
#include <algorithm>
using namespace std;
#define db double
const int M = 2005;
int read()
{
    int x=0,flag=1;
    char c;
    while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
    while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
    return x*flag;
}
int n,h,a[M],cl[M],cr[M];db p,f[M][M][2][2];
void init()
{
    cl[1]=1;
    for(int i=2;i<=n;i++)
    {
        if(a[i]-a[i-1]<h) cl[i]=cl[i-1];
        else cl[i]=i;
    }
    cr[n]=n;
    for(int i=n-1;i>=1;i--)
    {
        if(a[i+1]-a[i]<h) cr[i]=cr[i+1];
        else cr[i]=i;
    }
}
db dp(int l,int r,int x,int y)
{
    if(l>r) return 0;
    if(f[l][r][x][y]) return f[l][r][x][y];
    db tmp=0;
    //左边向左倒
    tmp+=0.5*p*(min(a[l]-a[l-1]-x*h,h)+dp(l+1,r,0,y));
    //右边向右倒
    tmp+=0.5*(1-p)*(min(a[r+1]-a[r]-y*h,h)+dp(l,r-1,x,0));
    //左边向右倒
    if(cr[l]>=r)
        tmp+=0.5*(1-p)*(min(a[r+1]-a[r]-y*h,h)+a[r]-a[l]);
    else
        tmp+=0.5*(1-p)*(dp(cr[l]+1,r,1,y)+a[cr[l]]-a[l]+h);
    //右边向左倒
    if(cl[r]<=l)
        tmp+=0.5*p*(min(a[l]-a[l-1]-x*h,h)+a[r]-a[l]);
    else
        tmp+=0.5*p*(dp(l,cl[r]-1,x,1)+a[r]-a[cl[r]]+h);
    return f[l][r][x][y]=tmp;
}
int main()
{
    n=read();h=read();
    scanf("%lf",&p);
    for(int i=1;i<=n;i++)
        a[i]=read();
    sort(a+1,a+n+1);
    init();
    a[0]=-1e9;
    a[n+1]=1e9;
    printf("%.9lf\n",dp(1,n,0,0));
}