7.19T2

小 B 的树

题目背景及题意

小 B 有一颗树，它的形态与 OI 中的树相同，有 n 个节点，n-1 条边，每条边长度为 1 随着时间的流逝这棵树长大了，每条边有 50%的概率长度变为 2 

小 B 想知道这棵树期望的直径是多少，可是他太菜了并不会，所以来求教你

输入格式

第一行一个 n 代表树上点的个数

接下来 n-1 行，每行两个整数，表示一条边

输出格式

一个实数，表示直径的期望

数据范围

打包，总共 10 个包，第 i 个包里的 n=i*5+10

 

sol：学长题解写的很好，自己都能看懂，就是写起来很操蛋(有详细注释，那处写挂的ts了一年）

upd：感觉自己全部在fp

#include <bits/stdc++.h>
using namespace std;
typedef int ll;
inline ll read()
{
    ll s=; bool f=; char ch=' ';
    while(!isdigit(ch)) {f|=(ch=='-'); ch=getchar();}
    while(isdigit(ch)) {s=(s<<)+(s<<)+(ch^); ch=getchar();}
    return (f)?(-s):(s);
}
#define R(x) x=read()
inline void write(ll x)
{
    if(x<) {putchar('-'); x=-x;}
    if(x<) {putchar(x+''); return;}
    write(x/); putchar((x%)+'');
}
#define W(x) write(x),putchar(' ')
#define Wl(x) write(x),putchar('\n')
const int N=,M=;
int n,mx[N],sz[N];
/*
    mx[i]表示包括i在内的以i为根的子树中的最长链
    sz[i]表示以i为根的子树大小
*/
int tot=,Next[M],to[M],head[N];
double dp[N][M][M],f[][M][M][M],ans=0.0;
/*
    dp[i][j][k]表示i的子树，最大深度为j，直径<=k的概率 做的时候按照直径=k做，然后搞一遍前缀和即可
    f[i][j][k][l]表示在x的子树中前i个儿子的子树中，最长链为j,次长链为k，直径<=l的概率(因为是从dp转移过来的，直接带前缀和)
*/
inline void Link(int x,int y)
{
    Next[++tot]=head[x]; to[tot]=y; head[x]=tot;
}
inline void dfs(int x,int fat)
{
    int e,i,j,k,l,t;
    sz[x]=;
    for(e=head[x];e;e=Next[e]) if(to[e]!=fat)
    {
        dfs(to[e],x); sz[x]+=sz[to[e]];
    }
    for(i=;i<=sz[x]*;i++) f[t=][][][i]=1.0;
    for(e=head[x];e;e=Next[e]) if(to[e]!=fat)
    {
        int V=to[e];
        t^=;
        for(i=;i<=mx[V];i++) for(j=;j<=(sz[x]*);j++) dp[V][i][j]+=dp[V][i][j-];//前缀和
        //注意这里j要转移到sz[x]*2而不是sz[V]*2，虽然后面对于V而言没有数值，但在x转移是要用啊！！！！！
        for(i=;i<=mx[x];i++) for(j=;j<=i;j++) for(k=;k<=sz[x]*;k++)//最长链，次长链，直径(不一定要求一定直径最长)
        {
            double tmp=f[t^][i][j][k];
            f[t^][i][j][k]=; //要加入一个新儿子了
            for(l=;l<=mx[V];l++)//枚举V中的最长链长度
            {
                double val=dp[V][l][k]*0.5;//0.5是要分类转移

                if(l+>i) f[t][l+][i][k]+=tmp*val;
                else if(l+>j) f[t][i][l+][k]+=tmp*val;
                else f[t][i][j][k]+=tmp*val;

                if(l+>i) f[t][l+][i][k]+=tmp*val;
                else if(l+>j) f[t][i][l+][k]+=tmp*val;
                else f[t][i][j][k]+=tmp*val;
            }
        }
        mx[x]=max(mx[x],mx[V]+);
    }
//    printf("%d %d %lf %lf ",x,mx[x],f[t][mx[x]][0][mx[x]],f[t][mx[x]][0][mx[x]-1]);
    for(i=;i<=mx[x];i++) for(j=;j<=i;j++) for(k=sz[x]*;k>=;k--)
    {
        if(k!=) f[t][i][j][k]-=f[t][i][j][k-];
        dp[x][i][max(i+j,k)]+=f[t][i][j][k];
        f[t][i][j][k]=;
    }
//    printf("%lf %lf\n",dp[x][mx[x]][mx[x]],dp[x][mx[x]][mx[x]-1]);
}
int main()
{
    freopen("tree.in","r",stdin);
    freopen("tree.out","w",stdout);
    int i,j,x,y;
    R(n);
    for(i=;i<n;i++)
    {
        R(x); R(y); Link(x,y); Link(y,x);
    }
    dfs(,);
//    printf("%lf %lf %lf %lf %lf\n",dp[1][4][4],dp[1][5][5],dp[1][6][6],dp[1][7][7],dp[1][8][8]);
//    for(i=1;i<=n;i++) cout<<i<<' '<<mx[i]<<' '<<sz[i]<<endl;
    for(i=;i<=mx[];i++) for(j=;j<=n*;j++) ans+=dp[][i][j]*j;
    printf("%lf\n",ans);
    return ;
}