Ice_cream’s world II(HDU 2121)---无定根最小树形图(超级源点)

题目链接

题目描述

After awarded lands to ACMers, the queen want to choose a city be her capital. This is an important event in ice_cream world, and it also a very difficult problem, because the world have N cities and M roads, every road was directed. Wiskey is a chief engineer in ice_cream world. The queen asked Wiskey must find a suitable location to establish the capital, beautify the roads which let capital can visit each city and the project’s cost as less as better. If Wiskey can’t fulfill the queen’s require, he will be punishing.

输入格式

Every case have two integers N and M (N<=1000, M<=10000), the cities numbered 0…N-1, following M lines, each line contain three integers S, T and C, meaning from S to T have a road will cost C.

输出格式

If no location satisfy the queen’s require, you must be output “impossible”, otherwise, print the minimum cost in this project and suitable city’s number. May be exist many suitable cities, choose the minimum number city. After every case print one blank.

输入样例

3 1
0 1 1
4 4
0 1 10
0 2 10
1 3 20
2 3 30

输出样例

impossible
40 0

分析

题目大意是给定n个点和m条有向边，求以某点为根节点建立树形图的最小权值，如果不能建立则输出"impossible"，否则输出最小权值和根节点编号，有多种情况就输出编号最小的那个。

对于这道题我们需要建立一个超级源点n+1，并建立超级源点到n个点的边，权值为图权值和+1（避免从超级源点出发的边有2条及以上），然后通过朱刘算法求最小树形图即可，对于朱刘算法不了解的可以看看这篇博客，主要的算法流程是这样的：

1. 求各点最小入边权值in[i]。
2. 判断除根节点外是否所有点都有入边，有则继续3，无则无法建树形图，输出“poor snoopy”。
3. 缩点并统计环，如果有环就继续4，无环则表示已找到最小树形图，输出其权值即可。
4. 建新图，对于边<u,v>，如果点u和v不在一个环路呢，则边<u,v>的权值减小in[v]。
5. 继续从1-4，直到图中无环。

需要注意的是，在用朱刘算法不断缩点建新图过程中，每条边所连接的两个点都是在变化的，因此我们不能直接记录下根节点的编号，需要通过对应的边的编号来i，来得到根节点编号i-m-1（代码中点编号为1~n），以下是源代码。

源程序

#include <bits/stdc++.h>
#define MAXN 1005
#define MAXM 20005
#define INF 0x3f3f3f3f
using namespace std;
struct Edge{
	int u,v,w;
}edge[MAXM];
int pos,in[MAXN],id[MAXN],pre[MAXN],vis[MAXN];
int zhuliu(int root,int n,int m)
{
	int res=0;	//权值
	while(1){
		for(int i=1;i<=n;i++)	//初始化入边 
			in[i]=INF;
		for(int i=1;i<=m;i++){	//统计最小入边 
			int u=edge[i].u,v=edge[i].v,w=edge[i].w;
			if(w<in[v] && u!=v){
				pre[v]=u;
				in[v]=w;
				if(u==root)	//记录实际位置,v在变化但i不变所以用i记录点 
					pos=i;
			}
		} 
		in[root]=0; 
		for(int i=1;i<=n;i++)	//判断能否成树
			if(in[i]==INF)
				return -1;
		int cnt=0;	//环数
		memset(id,-1,sizeof(id));
		memset(vis,-1,sizeof(vis));
		for(int i=1;i<=n;i++){
			res+=in[i];
			int v=i;
			while(vis[v]!=i&&id[v]==-1&&v!=root){	//寻找有向环
				//三个终止条件：找到相同标记，节点已属其他还，遍历到根 
				vis[v]=i;
				v=pre[v];
			}
			if(id[v]==-1&&v!=root){	//有环
				id[v]=++cnt;
				for(int u=pre[v];u!=v;u=pre[u])
					id[u]=cnt; 
			}
		} 
		if(cnt==0)	//无环
			break;
		for(int i=1;i<=n;i++)	//标记非环中的点 
			if(id[i]==-1)
				id[i]=++cnt; 
		for(int i=1;i<=m;i++){	//建新图
			int u=edge[i].u,v=edge[i].v;
			edge[i].u=id[u];
			edge[i].v=id[v];
			if(id[u]!=id[v])	//不在一个环中
				edge[i].w-=in[v]; 
		} 
		n=cnt;
		root=id[root];
	} 
	return res;
}
int main()
{
	int n,m;
	while(~scanf("%d%d",&n,&m)&&(n+m)){
		int sum=0;	//边权值和 
		for(int i=1;i<=m;i++){	//建边 
			scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
			edge[i].u++;
			edge[i].v++;
			sum+=edge[i].w; 
		}
		sum++;
		for(int i=m+1;i<=m+n;i++){	//超级源点
			edge[i].u=n+1;
			edge[i].v=i-m;
			edge[i].w=sum;
		}
		int res=zhuliu(n+1,n+1,m+n);
		if(res==-1 || res-sum>=sum)
			printf("impossible\n");
		else
			printf("%d %d\n",res-sum,pos-m-1);
		printf("\n");
	}
}