HDU 5723 Abandoned country（最小生成树Kruskal+树形DP）

Description
给一个n个点m条边的无向图，求最小生成树，输出最小生成树的权值和这个最小生成树任意两点间距离的平均值
Input
第一行一整数T表示用例组数，每组用例首先输入两个整数n和m分别表示点数和边数，之后m行每行三个整数u,v,w表示u和v之间存在一条权值为w的边
(T<=10,n<=10^5,m<=10^6,0<=w<=10^5)
Output
对于每组用例，输出最小生成树权值和以及这个最小生成树任意两点间距离的平均值，后者保留两位小数
Sample Input
1
4 6
1 2 1
2 3 2
3 4 3
4 1 4
1 3 5
2 4 6
Sample Output
6 3.33
Solution
求最小生成树套模版即可，而生成树上任意两点距离均值即为（每条边的权值*这条边两边的节点数乘积）/ C(n,2)，求某条边(u,v)两边的节点数可以任取一个根节点后将最小生成树转化为一棵有根树，对这棵有根树做树形DP即可得到以每个点作为根节点的子树节点总数，那么(u,v)两边的节点数即为min(cnt[u],cnt[v])，n-min(cnt[u],cnt[v])
注意最小生成树权值和可能会爆int
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 111111
#define maxm 1111111
struct node
{
    int u,v,w;
}edge[maxm];
int T,n,m,fa[maxn],flag[maxm],cnt[maxn];
vector<int>g[maxn];
int cmp(node a,node b)
{
    if(a.w!=b.w)return a.w<b.w;
    if(a.u!=b.u)return a.u<b.u;
    return a.v<b.v;
}
int find(int x)
{
    if(fa[x]==x) return x;
    return fa[x]=find(fa[x]);
}
ll kruskal()
{
    for(int i=1;i<=n;i++)fa[i]=i,g[i].clear();
    int cnt=0;
    ll ans=0;
    sort(edge,edge+m,cmp);
    for(int k=0;k<m;k++)
    {
        int x=find(edge[k].u),y=find(edge[k].v);
        if(x!=y)
        {
            cnt++;
            flag[k]=1;
            g[edge[k].u].push_back(edge[k].v);
            g[edge[k].v].push_back(edge[k].u);
            fa[x]=y;
            ans+=edge[k].w;
            if(cnt==n-1)return ans;
        }
    }
    return -1;
}
void dfs(int u,int fa)
{
    cnt[u]=1;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(v==fa)continue;
        dfs(v,u);
        cnt[u]+=cnt[v];
    }
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
            scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
        memset(flag,0,sizeof(flag));
        ll ans=kruskal();
        dfs(1,0);
        double w=0;
        for(int i=0;i<m;i++)
            if(flag[i])
            {
                int res=min(cnt[edge[i].u],cnt[edge[i].v]);
                w+=1.0*res*(n-res)*edge[i].w;
            }
        w=w*2/n/(n-1);
        printf("%I64d %.2lf\n",ans,w);
    }
    return 0;
}