poj 3164 Command Network【最小树形图】

题目大意：求最小树形图的总权值。即以固定根为起点，沿给定有向边，可以访问到所有的点，并使所构成的边权值之和最小，求这个最小总权值。
分析：朱刘算法。
^_^

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int INF=0x3f3f3f3f;
const int maxn=1010;
struct point{
    double x,y;
}p[maxn];
struct node{
    int u,v;
    double len;
}e[maxn*maxn];
int n,m;
double dist(point A,point B){
    return sqrt((A.x-B.x)*(A.x-B.x)+(A.y-B.y)*(A.y-B.y));
}
double in[maxn];
int pre[maxn],id[maxn],vis[maxn],be[maxn];
double solve(){
    double ans=0;
    int root=1;
    while(1){
        for(int i=1;i<=n;i++) in[i]=INF;
        for(int i=1;i<=m;i++){
            int u=e[i].u,v=e[i].v;
            if(u!=v && e[i].len<in[v]) in[v]=e[i].len,pre[v]=u;
        }
        for(int i=1;i<=n;i++)
            if(i==root) continue;
            else if(in[i]==INF) return -1;
        memset(id,-1,sizeof(id));
        memset(vis,-1,sizeof(vis));
        in[root]=0;
        int cnt=0;
        for(int i=1;i<=n;i++){
            ans+=in[i];
            int v=i;
            while(v!=root && id[v]==-1 && vis[v]!=i){
                vis[v]=i;
                v=pre[v];
            }
            if(v!=root && id[v]==-1){
                cnt++;
                for(int u=pre[v];u!=v;u=pre[u]) id[u]=cnt;
                id[v]=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=e[i].u,v=e[i].v;
            e[i].u=id[u];
            e[i].v=id[v];
            if(id[u]!=id[v])e[i].len-=in[v];
        }
        n=cnt;
        root=id[root];
    }
    return ans;
}
int main(){
    int i,j,k;
    while(scanf("%d%d",&n,&m)!=EOF){
        for(i=1;i<=n;i++) scanf("%lf%lf",&p[i].x,&p[i].y);
        for(i=1;i<=m;i++){
            scanf("%d%d",&e[i].u,&e[i].v);
            if(e[i].u==e[i].v) e[i].len=(double)INF;
            else e[i].len=dist(p[e[i].u],p[e[i].v]);
        }
        double ans=solve();
        if(ans==-1) printf("poor snoopy\n");
        else printf("%.2f\n",ans);
    }
    return 0;
}