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

[题目链接]

[题意]
n座城市m条路，每条路连接2个城市，路的长度各不相同，修建一条路需要的代价等于其长度。求:
1.使所有城市连通需要的最小代价
2.在最小代价下，任取2个城市，求他们之间最短路的期望值

[分析]
1.是比较裸的MST，直接上kruskal
由于边权各不相同，MST的形态应该是固定的
2.用树形DP求出所有点对的路径和，再除以点对的个数n*(n-1)/2，就是期望值
一条边被统计的次数等于两边点的个数的乘积

[代码]

#include <bits/stdc++.h>
using namespace std ;
const int N = 1e5 + 5 ;
typedef long long LL ;

int T , n , m ;

struct Edge
{
    int u , v , w ;
    void read() { scanf( "%d%d%d" , &u , &v , &w ) ; }
    bool operator < ( const Edge &rhs ) const
    {
        return w < rhs.w ;
    }
} E[N*10] ;

int head[N] , C ;
int f[N] , sz[N] ;
LL sum ;

struct edge
{
    int v , w , next ;
} e[N*2] ;

int find( int x )
{
    return x == f[x] ? x : f[x] = find(f[x]) ;
}

void init()
{
    C = 0 ;
    memset(head,-1,sizeof(head)) ;
    for( int i = 1 ; i <= n ; i++ )
        f[i] = i ;
}

void addEdge( int u , int v , int w )
{
    e[C].v = v ;
    e[C].w = w ;
    e[C].next = head[u] ;
    head[u] = C++ ;
}

LL kruskal()
{
    sort(E,E+m) ;
    LL ret = 0 ;
    int cnt = 1 ;
    for( int i = 0 ; i < m ; i++ )
    {
        int u = E[i].u , v = E[i].v , w = E[i].w ;
        int fu = find(u) , fv = find(v) ;
        if( fu != fv )
        {
            f[fu] = f[fv] ;
            addEdge(u,v,w) ;
            addEdge(v,u,w) ;
            ret += w ;
            cnt++ ;
        }
        if( cnt >= n ) break ;
    }
    return ret ;
}

void dfs( int u , int p )
{
    sz[u] = 1 ;
    for( int i = head[u] ; ~i ; i = e[i].next )
    {
        int v = e[i].v , w = e[i].w ;
        if( v == p ) continue ;
        dfs(v,u) ;
        sz[u] += sz[v] ;
        sum += 1LL*w*sz[v]*(n-sz[v]) ;
    }
}

int main()
{
    scanf( "%d" , &T ) ;
    while( T-- )
    {
        scanf( "%d%d" , &n , &m ) ;
        for( int i = 0 ; i < m ; i++ ) E[i].read() ;
        init() ;
        LL ans = kruskal() ;
        sum = 0 ;
        dfs(1,0) ;
        double avg = 0 ;
        if( n > 1 ) avg = 2.0*sum/n/(n-1) ;
        printf( "%I64d %.2f\n" , ans , avg ) ;
    }
    return 0 ;
}