POJ 3164 最小树形图模板

版权声明：本博客内容基本为原创，如有问题欢迎联系，转载请注明出处 https://blog.csdn.net/qq_41955236/article/details/82526120

题目链接：http://poj.org/problem?id=3164

题意：

       给你n个城市的坐标，m条边，每条边会给你一个单向的路，路径的权值就是两个城市的坐标上的距离，要你求一条从城市1开始，能走过所有城市的一条最短的路。

做法：

       有向图的最小生成树，也就是最小树形图的模板了，以城市1为根找一条能连通所有城市的路（单向）。具体的实现过程可以搜索一下朱刘算法，但是大部分都是只讲实现过程而不讲原理的。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<cmath>

using namespace std;
const int inf=0x3f3f3f3f;
int vis[105],idx[105],pre[1005],cnt;
double x[105],y[105],in[105];
struct node{
	int u,v;
	double w;
}edge[10005];
double solve(int t,int n,int m){
	int x,y,i,j;
	double ans=0;
	while(1){
		for(i=1;i<=n;i++)in[i]=(double)inf;
		for(i=1;i<=m;i++){
			node e=edge[i];
			if(e.u!=e.v&&e.w<in[e.v]){
				in[e.v]=e.w;
				pre[e.v]=e.u;
			}
		}
		for(i=1;i<=n;i++){
			if(in[i]==inf&&i!=t)return -1;
			if(i!=t)ans+=in[i];
		}
		int tot=0;
		memset(vis,-1,sizeof(vis));
		memset(idx,-1,sizeof(idx));
		for(i=1;i<=n;i++){
			y=i;
			while(vis[y]!=i&&idx[y]==-1&&y!=t){
				vis[y]=i;
				y=pre[y];
			}
			if(y!=t&&idx[y]==-1){
				tot++;
				idx[y]=tot;
				for(x=pre[y];x!=y;x=pre[x])idx[x]=tot;
			}
		}
		if(!tot)break;
		for(i=1;i<=n;i++)if(idx[i]==-1)idx[i]=++tot;
		for(i=1;i<=m;i++){
			int y=edge[i].v;
			edge[i].u=idx[edge[i].u];
			edge[i].v=idx[edge[i].v];
			if(edge[i].u!=edge[i].v)edge[i].w-=in[y];
		}
		n=tot;
		t=idx[t];
	}
	return ans;
}
double getdis(int a,int b){
    return sqrt((x[a]-x[b])*(x[a]-x[b])+(y[a]-y[b])*(y[a]-y[b]));
}
int main(){
	int n,m;
	while(scanf("%d%d",&n,&m)!=EOF){
		int i,j;
		cnt=0;
		for(i=1;i<=n;i++)scanf("%lf%lf",&x[i],&y[i]);
		for(i=1;i<=m;i++){
			int u,v;
			scanf("%d%d",&u,&v);
			if(u==v){
				m--;
				i--;
				continue;
			}
			edge[i].u=u;
			edge[i].v=v;
			edge[i].w=getdis(u,v);
		}
		double ans=solve(1,n,m);
		if(ans<0)printf("poor snoopy\n");
		else printf("%.2f\n",ans);
	}
	return 0;
}