FJ省队集训最终测试 T2

时间:2023-03-08 20:02:51

FJ省队集训最终测试 T2

FJ省队集训最终测试 T2

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);
}