POJ 3614 最小树形图

传送门就不给了，就是POJ3614，就是任性一回！
视频题解戳这里！

这题就是一个最小树形图模板题，所以我只放模板，想了解算法过程的可以看视频题解！

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e5+7;
int n,m,pre[maxn],id[maxn];
int vis[maxn];
double in[maxn];
struct point
{
	double x,y;
}p[maxn];
struct edge
{
	int from;
	int to;
	double cost;
}e[maxn];
double dis(point a,point b)
{
	return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
}
double zhu_liu(int root,int nv,int ne)
{
	double ans = 0;
	while(true)
	{
		for(int i=0;i<nv;i++) in[i] = INF;
		for(int i=0;i<ne;i++)
		{
			int u = e[i].from;
			int v = e[i].to;
			if(e[i].cost<in[v] && u!=v)
			{
				pre[v] = u;
				in[v] = e[i].cost;
			}
		}
		for(int i=0;i<nv;i++)
		{
			if(i==root) continue;
			if(in[i]==INF) return -1;
		}
		int cnt = 0;
		memset(id,-1,sizeof(id));
		memset(vis,-1,sizeof(vis));
		in[root] = 0;
		for(int i=0;i<nv;i++)
		{
			ans += in[i];
			int v = i;
			while(vis[v]!=i && id[v]==-1 && v!=root)
			{
				vis[v] = i;
				v = pre[v];
			}
			if(v!=root && id[v]==-1)
			{
				for(int u=pre[v];u!=v;u=pre[u])
				{
					id[u] = cnt;
				}
				id[v] = cnt++;
			}
		}
		if(cnt==0) break;
		for(int i=0;i<nv;i++)
		{
			if(id[i]==-1)
			{
				id[i] = cnt++;
			}
		}
		for(int i=0;i<ne;i++)
		{
			int v = e[i].to;
			e[i].from = id[e[i].from];
			e[i].to = id[e[i].to];
			if(e[i].from!=e[i].to)
			{
				e[i].cost -= in[v];
			}
		}
		nv = cnt;
		root = id[root];
	}
	return ans;
}
int main()
{
	while(~scanf("%d%d",&n,&m))
	{
		for(int i=0;i<n;i++) scanf("%lf%lf",&p[i].x,&p[i].y);
		for(int i=0;i<m;i++)
		{
			scanf("%d%d",&e[i].from,&e[i].to);
			e[i].from--;
			e[i].to--;
			if(e[i].from!=e[i].to) e[i].cost = dis(p[e[i].from],p[e[i].to]);
			else e[i].cost = INF;
		}
		double ans = zhu_liu(0,n,m);
		if(ans==-1) printf("poor snoopy\n");
		else printf("%.2lf\n",ans);
	}
	return 0;
}