题意:求一个树上任意两点距离的平均值。
思路:如果枚举任意两个点来求,一定会超时,不如换个思路,求每个边的贡献次数边权值的和就是任意两点距离的总和。然后除以总路径数之和n(n-1).
而每个边的贡献次数等于每个边两边顶点个数的乘积(这个自己举个例子),然后这题就可以转换成在一个有根树里,求每个顶点构成子树的节点个数,一遍dfs就可以得到,dfs写法见代码.选根这题,没有特定要求,可以任取。最后注意这题有坑,精度问题,int 会爆炸。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=10000+10;
ll sum[maxn],dp[maxn];
int n;
struct node{
int to,w;
};
vector<node>E[maxn];
void dfs(int root,int father)
{
sum[root]=1;
for(int i=0;i<E[root].size();i++){
int son=E[root][i].to;
int len=E[root][i].w;
if(son==father) continue;
dfs(son,root);
sum[root]+=sum[son];
dp[root]+=dp[son]+(sum[son]*(n-sum[son]))*len;
}
}
int main()
{
int T;scanf("%d",&T);
while(T--){
scanf("%d",&n);
for(int i=0;i<=n;i++) E[i].clear();
for(int i=1;i<n;i++){
int a,b,c;scanf("%d %d %d",&a,&b,&c);
E[a].push_back(node{b,c});
E[b].push_back(node{a,c});
}
memset(dp,0,sizeof(dp));
memset(sum,0,sizeof(sum));
dfs(0,-1);
int s=n*(n-1)/2;
double tem=dp[0];
printf("%.11f\n",((double)dp[0])/(s*1.0));
}
return 0;
}