Hdu 5866 Lucky E(树上点期望的最大值)

传送门:Hdu 5866 Lucky E

---

题意:给你一棵树,有n个节点,m种操作,树上的每个点相当于一个士兵,有一个伤值,
m中操作,每种操作都有一个点的标号和一个概率,表示可以对这个点的子树造成1点伤害,
求受伤最高的士兵的伤值的期望

---

题解:摘自http://www.cnblogs.com/duoxiao/p/5784466.html

考虑到操作数M较小而树较大，所以可以对操作数进行计算。
假设初始时树上的节点的最大值为x，则在操作完成后的树上的节点的最大值不超过x+M,不低于x，所以可以枚举最终状态的值再计算概率。
设dp[i][k]表示操作i所作用的那棵子树v在操作完成后最大值不超过x+k的概率,maxtree(v) 为该子树初始时的最大值
那么可得递推式
dp[i][k]=p[i] ×× π(dp[j][k-1])+(1-p[i]) ×× π(dp[j][k-1]) j为i的同一棵子树上的后继操作( maxtree(v)<=maxtree(root)+k);
or dp[i][k]=0;(maxtree(v)>maxtree(root)+k)；
注意到子树的最大值会比maxtree(root)小很多，因此k可能为负， 所以要加上偏移量M
为了便于计算，添加作用于根节点的概率为0的操作(root,0)
然后预处理出每个操作u的后继操作即可。
先给每一个树标记上dfs的入序indfn[v]和出序outdfn[v]，那么作用在v节点上的操作的影响区间就是[indfn[v],outdfn[v]], 所以计算出区间之间的包含关系即可
最后结果为Σ（maxtree(root)+k) ×× (dp[0][k+MAX_M]-dp[0][k-1+MAX_M]);

#include<bits/stdc++.h>
using namespace std;
const int maxn=100100;
typedef pair<int,double> P;
const int MAX_Q=3011;
vector<int>G[maxn],mp[maxn];
double dp[MAX_Q][MAX_Q*2];
int vis[maxn],root,st[maxn],ed[maxn],tot,maxv[maxn],n,m;
P Q[maxn];

bool cmp(P x,P y){
    if(st[x.first]!=st[y.first])
        return st[x.first]<st[y.first];
    return ed[x.first]>ed[y.first];
}

void dfs(int u){
    st[u]=++tot;
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i];
        dfs(v);
        maxv[u]=max(maxv[u],maxv[v]);
    }
    ed[u]=tot;
}

int getnext(int u,int pre){
    int v=Q[u].first;
    if(~pre)    mp[pre].push_back(u);
    for(int i=u+1 ; i<=m; i=getnext(i,u))  //计算出每个点的子树修改的范围
        if(st[Q[i].first] > ed[v] )
            return i;
    return m+1;
}

double cal(int pre,int k){
    int v=Q[pre].first;
    if(dp[pre][k+MAX_Q]>-0.5)   return dp[pre][k+MAX_Q];
    if(maxv[v] > maxv[root]+k)    return 0;
    double ans1=Q[pre].second,ans2=1.0-Q[pre].second;
    if(maxv[v]==maxv[root]+k)   ans1=0;
    for(int i=0;i<mp[pre].size();i++){  //这个覆盖的查询的范围
         ans1*=cal(mp[pre][i],k-1);
         ans2*=cal(mp[pre][i],k);
    }
    dp[pre][k+MAX_Q]=ans1+ans2;
    return dp[pre][k+MAX_Q];
}

int main(){
    while(scanf("%d%d",&n,&m)!=EOF){
        int u,v;
        memset(dp,-1,sizeof(dp));
        memset(vis,0,sizeof(vis));
        for(int i=0;i<=n;i++)
            mp[i].clear(),G[i].clear();
        for(int i=1;i<n;i++)
            scanf("%d%d",&u,&v),vis[u]=1,G[v].push_back(u);
        for(int i=1;i<=n;i++){
            scanf("%d",&maxv[i]);
            if(vis[i]==0)   root=i;
        }
        tot=0;
        dfs(root);
        for(int i=1;i<=m;i++)
             scanf("%d%lf",&Q[i].first,&Q[i].second);
        Q[0]={root,0};
        sort(Q+1,Q+m+1,cmp);
        getnext(0,-1);
        double ans=0;
        for(int i=0;i<=m;i++)
            ans+=1.0*(maxv[root]+i)*( cal(0,i) - cal(0,i-1)  );
        printf("%.6f\n",ans);
    }
    return 0;
}