HDU - 4219 Randomization?（树形dp，概率）

题目：给你一棵n个节点的树(n<=64)，每条边都会随机的从0~L中间取一个边权值(L<=8)，求问生成的树任意两点间距离不超过S(S<=512)的概率是多少？

最后还是看了别人的思路，我好菜。。。

dp[u][i] 表示以u为子树，子树上两两节点距离不超过 s，且根到叶子节点的最长长度为 i 的概率。有了这个思路就都好说了。最后求得的就是 sigma(dp[1][i]) (0<=i<=s); dp表示的含义太神奇了，非常巧妙的dp方程。

#include <bits/stdc++.h>

using namespace std;
double dp[66][666];
vector<int>a[66];
int t,n,l,s;
double tmp[666],tmq[666];
void dfs(int u,int fa)
{
    dp[u][0]=1.0;//开始初始值是u节点选择的最长为0的路时，两两距离<=s的概率为1
    for(int i=0;i<a[u].size();i++)
    {
        int v=a[u][i];
        if(v==fa) continue;
        dfs(v,u);

        memset(tmp,0,sizeof tmp);
        memset(tmq,0,sizeof tmq);
        for(int j=0;j<=l;j++)
        for(int k=0;k<=s;k++)
        if(j+k<=s)
            tmq[j+k]+=dp[v][k]*1.0/(1.0*l+1.0);//从子节点过来的转移方程，tmp记录当前这个子节点v过来后连上u后的概率

        for(int j=0;j<=s;j++)
        for(int k=0;k+j<=s;k++)
        tmp[max(j,k)]+=dp[u][k]*tmq[j];//tmp[i]暂时代表合并完当前v后的dp[u][i]的值
		
		for(int j=0;j<=s;j++)
        dp[u][j]=tmp[j];
    }
}


int main()
{
    ///freopen("in.txt","r",stdin);
    scanf("%d",&t);
    int cas=0;
    while(t--)
    {
        scanf("%d%d%d",&n,&l,&s);
        for(int i=1;i<=n;i++)
            a[i].clear();
        for(int i=1;i<n;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            a[x].push_back(y);
            a[y].push_back(x);
        }
        memset(dp,0,sizeof dp);
        dfs(1,-1);
        double ans=0.0;
        for(int i=0;i<=s;i++)
            ans+=dp[1][i];///累加1的最远距离为i时，子节点两两距离不超过s的概率
        printf("Case %d: %.6lf\n",++cas,ans);
    }
    return 0;
}