POJ 3164 Command Network（最小树形图）

Description

给出n个点的坐标和m条有向边，边权为两点间距离，求这张图的最小树形图

Input

多组用例，每组用例第一行为两个整数n和m表示点数和边数，之后n行每行两个整数x和y表示该点坐标，之后m行每行两个整数a和b表示点a到点b之间有一条有向边，以文件尾结束输入(n<=100,m<=10000)

Output

对于每组用例，如果该图以1为起点存在最小树形图则输出最小树形图的权值和，否则输出”poor snoopy”

Sample Input

4 6

0 6

4 6

0 0

7 20

1 2

1 3

2 3

3 4

3 1

3 2

4 3

0 0

1 0

0 1

1 2

1 3

4 1

2 3

Sample Output

31.19

poor snoopy

Solution

最小树形图裸题

Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
#define INF 0x3f3f3f3f
#define maxn 111
#define maxm 22222
struct Edge
{
int u,v;
double c;
}edge[maxm];
int n,m,pre[maxn],id[maxn],vis[maxn];
double in[maxn];
double zhuliu(int root)
{
double ans=0;
int u,v;
while(1)
{
for(int i=0;i<n;i++)in[i]=INF;
for(int i=0;i<m;i++)
if(edge[i].u!=edge[i].v&&edge[i].c<in[edge[i].v])
{
pre[edge[i].v]=edge[i].u;
in[edge[i].v]=edge[i].c;
}
for(int i=0;i<n;i++)
if(i!=root&&in[i]==INF)
return -1;
int res=0;
memset(vis,-1,sizeof(vis));
memset(id,-1,sizeof(id));
in[root]=0;
for(int i=0;i<n;i++)
{
ans+=in[i];
v=i;
while(vis[v]!=i&&id[v]==-1&&v!=root)
{
vis[v]=i;
v=pre[v];
}
if(id[v]==-1&&v!=root)
{
for(u=pre[v];u!=v;u=pre[u])
id[u]=res;
id[v]=res++;
}
}
if(res==0)break;
for(int i=0;i<n;i++)
if(id[i]==-1)
id[i]=res++;
for(int i=0;i<m;)
{
v=edge[i].v;
edge[i].u=id[edge[i].u];
edge[i].v=id[edge[i].v];
if(edge[i].u!=edge[i].v)
edge[i++].c-=in[v];
else swap(edge[i],edge[--m]);
}
n=res;
root=id[root];
}
return ans;
}
struct node
{
double x,y;
}p[maxn];
double dis(node a,node b)
{
return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
}
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++)
{
int a,b;
scanf("%d%d",&a,&b);
a--,b--;
edge[i].u=a,edge[i].v=b,edge[i].c=dis(p[a],p[b]);
}
double ans=zhuliu(0);
if(ans==-1)printf("poor snoopy\n");
else printf("%.2f\n",ans);
}
return 0;
}