求最小树形图,也称为最小有向生成树。
朱-刘算法的过程为:
预处理: 删除自环 ,并判断能否从根节点到达其他所有节点。
① 给非根节点找最小入边 。
②如果这些边不构成环 ,则找到最小树形图。否则把各圈缩成点,继续重复操作。
③缩圈后要重新标号,和修改一些边的权值,确保的到答案正确。
代码:
#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;
}