poj 3164 Command Network 最小树形图

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

题目大意：

有固定根的最小树形图求法O(VE)，首先消除自环，显然自环不在最小树形图中。

想说一下，这个代码的朱刘算法是从0—n-1的。

这个代码想  吐槽一下，我提交了十几次，居然是因为  用g++交的，所以不过，改用c++就过了，，哪位路过的大神  解释一下，不胜感激。。。

代码如下：

#include <cstdio>
#include<iostream>
#include <cstring>
#include <cstdlib>
#include<cmath>
using namespace std;
typedef double type;
#define INF 2000000000
#define N 105
struct point //点的结构体
{
    double x,y;
}p[N];

struct edge //图的结构体
{
    int u,v;
    double  w;
}e[N*N];

int m,n,pre[N],id[N],visit[N];
double in[N];
//eCnt为图中的边数
//n为图中的顶点数
//pre[i]为顶点i的前驱节点
//id[i]为缩环，形成新图的中间量
//in[i]为点i的最小入边
//visit[i]遍历图时记录顶点是否被访问过
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 directedMST(int root,int nv,int ne)
{
    double ans=0;
    while(1)
    {
        //1.找最小入边
        for(int i=0;i<nv;i++) in[i]=INF;
        for(int i=0;i<ne;i++)
        {
            int u=e[i].u;
            int v=e[i].v;
            if(u!=v&&e[i].w<in[v])
            {
                in[v]=e[i].w;
                pre[v]=u;
            }
        }
        for(int i=0;i<nv;i++)                          //判断图是否连通
            if(i!=root&&in[i]==INF) return -1;  //除了跟以外有点没有入边,则根无法到达它
         //2.找环
        int nodeCnt=0;          //图中环的数目
        memset(id, -1, sizeof(id));
        memset(visit, -1, sizeof(visit));
        in[root]=0;
        for(int i=0;i<nv;i++)
        {
            ans+=in[i];
            int v=i;
            while(visit[v]!=i&&id[v]==-1&&v!=root)//每个点寻找其前序点，要么最终寻找至根部，要么找到一个环
            {
                visit[v]=i;
                v=pre[v];
            }
            if(v!=root&&id[v]==-1)//缩点
            {
                for(int u=pre[v];u!=v;u=pre[u])
                    id[u]=nodeCnt;
                id[v]=nodeCnt++;
            }
        }
        if(nodeCnt==0) break;//如果无环，跳出循环
        for(int i=0; i<nv; i++)
            if(id[i]==-1)
                id[i]=nodeCnt++;
        //3.缩点，重新标记
        for(int i=0;i<ne;i++)
        {
            int v=e[i].v;
            e[i].u=id[e[i].u];
            e[i].v=id[e[i].v];
            if(e[i].u!=e[i].v)
                e[i].w-=in[v];
        }
        nv=nodeCnt;
        root=id[root];
    }
    return ans;
}

int main()
{
    int m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        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].u,&e[i].v);
            e[i].u--;e[i].v--;//上面的朱刘算法是从0-（n-1）的点
            if(e[i].u!=e[i].v)
                e[i].w=Dis(p[e[i].u],p[e[i].v]);
            else
                e[i].w=INF;//消除自环
        }
        double solve=directedMST(0,n,m);
        if(solve==-1) printf("poor snoopy\n");
        else printf("%.2lf\n",solve);
    }
    return 0;
}