FJ省队集训最终测试 T2

思路:发现如果一个人一共选了x个点，那么选中某一个点对的概率都是一样的，一个人选x个点的总方案是C(n,x)，一个人选中某个点对的总方案是C(n-2,x-2)，这样，那么选中某个点对的概率就是 x*(x-1)/（n*(n-1)），这样，我们就用树分治求出有多少对符合条件的对数，然后乘上每个人的概率即可。

 #include<cstdio>
 #include<iostream>
 #include<cmath>
 #include<cstring>
 #include<algorithm>
 #define inf 0x7fffffff
 int son[],F[],root,vis[],pd[],c[];
 int n,go[],tot,first[],next[],dis[],num,sz,m,b[];
 int ans,cnt[];
 double Ans;
 int read(){
     int t=,f=;char ch=getchar();
     while (ch<''||ch>''){if (ch=='-')f=-;ch=getchar();}
     while (''<=ch&&ch<=''){t=t*+ch-'';ch=getchar();}
     return t*f;
 }
 void insert(int x,int y){
     tot++;
     go[tot]=y;
     next[tot]=first[x];
     first[x]=tot;
 }
 void add(int x,int y){
     insert(x,y);insert(y,x);
 }
 void findroot(int x,int fa){
     son[x]=;F[x]=;
     for (int i=first[x];i;i=next[i]){
         int pur=go[i];
         if (vis[pur]||pur==fa) continue;
         findroot(pur,x);
         son[x]+=son[pur];
         F[x]=std::max(F[x],son[pur]);
     }
     F[x]=std::max(F[x],num-son[x]);
     if (F[x]<F[root]) root=x;
 }
 void bfs(int x){
     int h=,t=;c[h]=x;pd[x]=sz;dis[x]=;
     while (h<=t){
         int now=c[h++];
         for (int i=first[now];i;i=next[i]){
           int pur=go[i];
           if (vis[pur]||pd[pur]) continue;
           pd[pur]=sz;
           dis[pur]=dis[now]+;
           c[++t]=pur;
         }
     }
     for (int j=;j<=m;j++)
      for (int i=;i<=t;i++)
       if (b[j]>=dis[c[i]])
       ans+=cnt[b[j]-dis[c[i]]];
     for (int i=;i<=t;i++)
      cnt[dis[c[i]]]++;
 }
 void solve(int x,int fa){
     vis[x]=;sz++;
     memset(cnt,,sizeof cnt);cnt[]=;
     for (int i=first[x];i;i=next[i]){
         int pur=go[i];
         if (vis[pur]) continue;
         bfs(pur);
     }
     int Sum=num;
     for (int i=first[x];i;i=next[i]){
         int pur=go[i];
         if (vis[pur]) continue;
         if (son[pur]>son[x]) num=Sum-son[x];
         else num=son[pur];
         root=;
         findroot(pur,);
         solve(root,x);
     }
 }
 int main(){
     n=read();m=read();
     for (int i=;i<=m;i++){
         b[i]=read();
     }
     std::sort(b+,b++m);
     for (int i=;i<n;i++){
         int u=read(),v=read();
         add(u,v);
     }
     F[]=inf;root=;num=n;
     findroot(,);
     solve(root,);
     double Ans=(((double)ans)/((double)n))/((double)n-);
     int m=n/;
     if (n%) printf("%.2lf\n",Ans*(m+)*(m));
     else printf("%.2lf\n",Ans*(m-)*m);
     if (n%==) printf("%.2lf\n",Ans*(m+)*(m));
     else printf("%.2lf\n",Ans*(m-)*m);
     printf("%.2lf\n",Ans*(m-)*m);
 }