bzoj 4675 点对游戏 - 长链剖分

题解：根据期望的线性性，可知答案是$\frac{\left(\genfrac{}{}{0ex}{}{n-2}{k-2}\right)}{\left(\genfrac{}{}{0ex}{}{n}{k}\right)}{\sum }_{1\le u<v\le n}\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(u,v)\in M$。后半部分直接长链剖分即可。

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define Rep(i,v) rep(i,0,(int)v.size()-1)
#define lint long long
#define ull unsigned lint
#define db long double
#define pb push_back
#define mp make_pair
#define fir first
#define sec second
#define gc getchar()
#define N 50010
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
typedef pair<int,int> pii;
typedef set<int>::iterator sit;
inline int inn()
{
    int x,ch;while((ch=gc)<'0'||ch>'9');
    x=ch^'0';while((ch=gc)>='0'&&ch<='9')
        x=(x<<1)+(x<<3)+(ch^'0');return x;
}
struct edges{
    int to,pre;
}e[N<<1];int h[N],etop,l[N],son[N];
inline int add_edge(int u,int v) { return e[++etop].to=v,e[etop].pre=h[u],h[u]=etop; }
int getl(int x,int fa=0)
{
    l[x]=0,son[x]=0;
    for(int i=h[x],y;i;i=e[i].pre)
        if((y=e[i].to)^fa)
        {
            getl(y,x),l[x]=max(l[x],l[y]+1);
            if(!son[x]||l[y]>l[son[x]]) son[x]=y;
        }
//  debug(x)sp,debug(l[x])sp,debug(son[x])ln;
    return 0;
}
int m,v[20];lint ans;
int dfs(int x,int *f,int fa=0)
{
    if(son[x]) dfs(son[x],f+1,x);else return f[0]=1;
    f[0]++;rep(i,1,m) if(v[i]<=l[x]) ans+=f[v[i]];
    for(int i=h[x],y;i;i=e[i].pre)
        if((y=e[i].to)!=son[x]&&e[i].to!=fa)
        {
            int *fy=new int[l[y]+1];
            memset(fy,0,sizeof(int)*(l[y]+1)),dfs(y,fy,x);
            for(int j=0;j<=l[y];j++) for(int k=1;k<=m;k++)
                if(v[k]-j-1>=0&&v[k]-j-1<=l[x]) ans+=(lint)f[v[k]-j-1]*fy[j];
            for(int j=0;j<=l[y];j++) f[j+1]+=fy[j];
        }
//  debug(x)ln;rep(i,0,l[x]) debug(i)sp,debug(f[i])ln;cerr ln;
    return 0;
}
int main()
{
    int n=inn(),x,y;m=inn();rep(i,1,m) v[i]=inn();
    rep(i,1,n-1) x=inn(),y=inn(),add_edge(x,y),add_edge(y,x);
    int k=n/3,k1=k,k2=k,k3=k;if(n%3>=1) k1++;if(n%3==2) k2++;
    getl(1);int *f=new int[l[1]+1];
    memset(f,0,sizeof(int)*(l[1]+1)),dfs(1,f);
    printf("%.2lf\n",(double)(k1*(k1-(db)1)/n/(n-(db)1)*ans));
    printf("%.2lf\n",(double)(k2*(k2-(db)1)/n/(n-(db)1)*ans));
    printf("%.2lf\n",(double)(k3*(k3-(db)1)/n/(n-(db)1)*ans));
    return 0;
}