HDU 5723-A - Abandoned country-最小生成树+树dp

按题意求一个最小生成树，然后走一遍dfs记录每个点为根的子树大小，那么走过 某条边的点对就是 num[x]*（n-num[x]），累加一下除边就好

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>
#include <queue>
#include <map>
#include <set>
#include <vector>
using namespace std;
struct node
{
    int x,y;
    int w;
    node() {}
    node(int a,int b,int c)
    {
        w=c;
        x=a;
        y=b;
    }
};
node tm[1123456];
int cmp(node a,node b)
{
    return a.w<b.w;
}
int fa[112345];
int find(int x)
{
    if (fa[x]==x)
        return fa[x];
    else
        return fa[x]=find(fa[x]);
}
vector<node > mp[112345];
int num[112345];
void dfs(int x,int fa)
{
    num[x]=1;
    for (int i=0; i<mp[x].size(); i++)
    {
        int v=mp[x][i].x;
        if (v==fa)continue;
        dfs(v,x);
        num[x]+=num[v];
    }
}

int n,m;
double ans=0;
void dfs2(int x,int fa)
{
    for (int i=0; i<mp[x].size(); i++)
    {
        int v=mp[x][i].x;
        if (v==fa)continue;
        double tmp=mp[x][i].y;
        tmp=tmp*num[v]*(n-num[v]);
        ans+= tmp;
        dfs2(v,x);
    }
}
void init()
{
    for (int  i=1; i<=n; i++)
        mp[i].clear(),fa[i]=i;
    ans=0;

}
int main()
{
    int tmp;
    int t;
    cin>>t;
    while(t--)
    {
        cin>>n>>m;
        init();
        int x,y,z;
        for (int i=1; i<=m; i++)
        {
            scanf("%d %d %d",&x,&y,&z);
            tm[i].x=x;
            tm[i].y=y;
            tm[i].w=z;
        }
        sort(tm+1,tm+1+m,cmp);

        long long sum=0;
        for (int i=1; i<=m; i++)
        {
            int x=tm[i].x;
            int y=tm[i].y;
            int fx=find(x);
            int fy=find(y);
            if (fx!=fy)
            {
                fa[fx]=fy;
                mp[x].push_back(node(y,tm[i].w,0));
                mp[y].push_back(node(x,tm[i].w,0));
                sum+=tm[i].w;
            }
        }
        dfs(1,0);
        dfs2(1,0);

        double tmp=n;
        printf("%lld %.2lf\n",sum,ans/(tmp*(tmp-1)/2));


    }

    return 0;

}