发现一个人如果最终拿走了(k)个点,那么这个人的答案就是
[frac{inom{n-2}{k-2}sum_{i=1}^{n}sum_{j=1}^{n}[dis(i,j)in M]}{inom{n}{k}}
]
考虑每一个点对对答案的贡献,我们枚举一个点对,之后对于剩下的(n-2)个点我们随便选出(k-2)对点组成我们所选择的(k)个点,只需要看看这个点对的距离是否属于(M)就好了
组合数划开发现是(frac{k imes(k-1)}{n imes (n-1)}),于是我们可以直接长链剖分求后面那个柿子就好了
代码
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#define re register
#define LL long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
const int maxn=50005;
inline int read() {
char c=getchar();int x=0;while(c<'0'||x>'9') c=getchar();
while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-48,c=getchar();return x;
}
struct E{int v,nxt;}e[maxn<<1];
int len[maxn],n,m,son[maxn];
int deep[maxn],head[maxn];
int tax[maxn],*id=tax,*f[maxn];
LL ans;
int num,a[15];
inline void add(int x,int y) {
e[++num].v=y;e[num].nxt=head[x];head[x]=num;
}
void dfs1(int x) {
for(re int i=head[x];i;i=e[i].nxt) {
if(deep[e[i].v]) continue;
deep[e[i].v]=deep[x]+1;
dfs1(e[i].v);
if(len[e[i].v]>len[son[x]]) son[x]=e[i].v;
}
len[x]=len[son[x]]+1;
}
void dfs(int x) {
f[x][0]=1;
if(son[x]) {
f[son[x]]=f[x]+1,dfs(son[x]);
for(re int k=1;k<=m;k++)
if(a[k]<len[x]) ans+=f[x][a[k]];
}
for(re int i=head[x];i;i=e[i].nxt) {
if(deep[e[i].v]<deep[x]||e[i].v==son[x]) continue;
f[e[i].v]=id;id+=len[e[i].v];dfs(e[i].v);
for(re int j=0;j<len[e[i].v];j++)
for(re int k=1;k<=m;k++)
if(a[k]-j-1>=0&&a[k]-j-1<len[x]) ans+=f[e[i].v][j]*f[x][a[k]-j-1];
for(re int j=0;j<len[e[i].v];j++)
f[x][j+1]+=f[e[i].v][j];
}
}
inline double C(double n,double m) {
double t=m*(m-1),b=n*(n-1);
return t/b;
}
int main() {
n=read(),m=read();
for(re int i=1;i<=m;i++) a[i]=read();
for(re int x,y,i=1;i<n;i++)
x=read(),y=read(),add(x,y),add(y,x);
deep[1]=1,dfs1(1);
f[1]=id;id+=len[1];dfs(1);
int k[3];int h=n,now=0;k[0]=k[1]=k[2]=0;
while(h) {k[now]++;now++;h--;now%=3;}
for(re int i=0;i<3;i++)
if(k[i]>=2) printf("%.2lf
",C(n,k[i])*ans);
else puts("0.00");
return 0;
}