hdu5723 Abandoned country 最小生成树 + dfs

题意：给定一张带权的无向图，先建立最小生成树，问在最小生成树中，任选两点的距离最小值的数学期望是多少？

思路：首先建立最小生成树，建好图，数学期望E = Σ 1/(n * (n - 1) / 2) * i和j之间的最小距离 = 1/(n * (n - 1) / 2) * Σe.to的子树中节点数*(cnt - e.to的子树中节点数 )*e.w （其中e是生成树的树边），其中每个点的子树中节点的个数可以dfs求出，再用一次dfs求出 “Σe.to的子树中节点数*(cnt - e.to的子树中节点数 )*e.w ”，代入E中得到答案。

可以看图来理解一下，红色的树边被经过了3 * （7 - 3）次，每次都要加上一个对应的权值e.w,所以是3 * (7 - 3) *e.w ，其他边亦然，所以总的结果就是把每条边对应的积累加起来。

代码：

#pragma warning(disable:4786)
#pragma comment(linker, "/STACK:102400000,102400000")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<cmath>
#include<string>
#include<sstream>
#define LL long long
#define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;++i)
#define mem(a,x) memset(a,x,sizeof(a))
#define lson l,m,x<<1
#define rson m+1,r,x<<1|1
using namespace std;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
const double PI = acos(-1.0);
const double eps=1e-6;
const int maxn = 1e5 + 5 ;
const int maxm = 1e6 + 5 ;
int n , m ;
int p[maxn] , fis;
vector<int>G[maxn] ;
double  ans = 0 ;
int cnt[maxn];
struct edge
{
    int from , to , w ;
    bool operator < (const edge &rhs)const
    {
        return w < rhs.w ;
    }
}edges[maxm];
void addedge(int x , int y , int num)
{
    G[x].push_back(num) ;
    G[y].push_back(num) ;
}
void init(int n)
{
    for(int i = 1 ; i<= n ; i++){
        p[i] = i;
        G[i].clear() ;
    }
    return ;
}
int find(int x)
{
    if(p[x] == x)
        return p[x] ;
    else
        return p[x] = find(p[x]) ;
}
void merge(int x , int y)
{
    int rootx = find(x) ;
    int rooty = find(y) ;
    if(rootx == rooty)
        return ;
    p[rootx] = rooty ;
}
LL kruskal()
{
    sort(edges+1 , edges + 1 + m) ;
    init(n);
    LL  ans = 0 , sum = 0;
    for(int i = 1 ; i<= m ; i++){
        int x = edges[i].from ;
        int y = edges[i].to ;
        int w = edges[i].w;
        if(find(x) == find(y))
            continue;
        if(fis == -1)
            fis = x ;
        ans += w;
        merge(x , y) ;
        addedge(x , y , i) ;
        ++sum ;
        if(sum == n - 1 )
            return ans;
    }
}
int dfs1(int cur , int fa)
{
    int len = G[cur].size();
    ++cnt[cur] ;
    for(int i = 0 ; i<len ; i++){
        edge &e = edges[G[cur][i]] ;
        int nt  ;
        if(e.from == cur)
            nt = e.to;
        else
            nt = e.from ;
        if(nt == fa)    continue ;
        cnt[cur]  +=  dfs1(nt , cur)  ;
    }
    return cnt[cur];
}
void dfs2(int cur , int fa)
{
    int len = G[cur].size();
    for(int i = 0 ; i<len ; i++){
        edge &e = edges[G[cur][i]] ;
        int nt  ;
        if(e.from == cur)
            nt = e.to;
        else
            nt = e.from ;
        if(nt == fa)    continue ;
        ans += (double)cnt[nt] * (double) (n  - cnt[nt]) * (double) e.w ;
        dfs2(nt , cur) ;
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n , &m) ;
        for(int i = 1 ; i<= m ; i++){
            scanf("%d%d%d",&edges[i].from, &edges[i].to , &edges[i].w) ;
        }
        fis = -1  ;
        LL res = kruskal();
        printf("%lld ",res) ;
        mem(cnt , 0) ;
        ans = 0;
        dfs1(fis , -1) ;
        dfs2(fis , -1) ;
        double k = (double)n *  ((double)n - 1) * (double ) 0.5  ;
        ans = 1.0 * ans / k ;
        printf("%.2lf\n",ans);
    }
    return 0;
}