uva 11183(最小树形图)

求最小树形图，也称为最小有向生成树。

朱-刘算法的过程为：

预处理：  删除自环 ，并判断能否从根节点到达其他所有节点。

① 给非根节点找最小入边 。

②如果这些边不构成环 ，则找到最小树形图。否则把各圈缩成点，继续重复操作。

③缩圈后要重新标号，和修改一些边的权值，确保的到答案正确。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int INF = 1e9 ;
const int maxn = 1010 , maxe = 40050 ;

struct Edge{int from ,to , dist ; };
Edge edges[maxe] ;
int n , m ;
int id[maxn] , in[maxn] , pre[maxn] , vis[maxn];

int directed_mst(int root){
    int ans = 0 ;
    while(true)
    {
        // ①找最小入边----------------------------
        for(int i=0 ; i<n ;i++) in[i] = INF ;
        for(int i=0 ; i<m ;i++)
        {
            int u = edges[i].from , v = edges[i].to , w = edges[i].dist ;
            if(w < in[v] && u!=v){
                in[v] = w ; pre[v] = u ;
            }
        }
        for(int i=0 ;i<n ;i++)
        {
            if(i == root) continue ;
            if(in[i] == INF) return -1 ; // 图不连通，不存在树形图
        }

        //②找环-----------------------------
        int cnt_node = 0; //  缩圈后节点个数
        memset(id , -1 ,sizeof(id));
        memset(vis , -1 ,sizeof(vis));
        in[root] = 0;
        for(int i=0 ;i<n ;i++)  // 先给环上的点编号
        {
            ans += in[i] ;
            int p = i ;
            while(vis[p] !=i && id[p] == -1 && p!=root)
            {
                vis[p] = i;  p = pre[p] ;
            }
            if(id[p] == -1 && p!=root)  // 只可能是vis[p] == i ,  即找到了环
            {
                for(int u=pre[p] ; u!=p ; u = pre[u]){
                    id[u] = cnt_node ;
                }
                id[p] = cnt_node++ ;
            }
        }
        if(cnt_node == 0) break;  // 没有环了， 找到最小树形图！
        for(int i=0 ;i<n ;i++)
        {
            if(id[i] == -1) id[i] = cnt_node++ ;
        }

        // ③缩点 ， 重新编号---------------------------
        for(int i=0 ;i<m ;i++)
        {
            int v = edges[i].to ;
            edges[i].from = id[edges[i].from] ;
            edges[i].to = id[edges[i].to] ;
            if(edges[i].from != edges[i].to)
            {
                edges[i].dist -= in[v] ;
            }
        }
        n = cnt_node  ;
        root = id[root] ;
    }
    return ans ;
}

int main()
{
    //freopen("in.txt" ,"r",stdin) ;
    int T , cas = 0 ;
    scanf("%d" ,&T) ;
    while(T--)
    {
        cas++ ;
        scanf("%d%d" ,&n ,&m) ;
        for(int i=0 ;i<m ;i++)
            scanf("%d%d%d" ,&edges[i].from , &edges[i].to ,&edges[i].dist) ;
        int ans = directed_mst(0) ;
        printf("Case #%d: " , cas) ;
        if(ans == -1) printf("Possums!\n") ;
        else printf("%d\n" , ans);
    }
    return 0;
}