poj 3164 最小树形图

思路：就是裸的最小树形图~

#include<iostream>

#include<cstdio>

#include<cstring>

#include<cmath>

#include<algorithm>

#define Maxn 110

#define LL double

#define inf 1e9

using namespace std;

int pre[Maxn],id[Maxn],vi[Maxn],num;

LL in[Maxn];

struct Edge{

    int u,v;

     LL val;

}edge[Maxn*Maxn];

struct Point{

    double x,y;

}p[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));

}

void add(int u,int v,LL val)

{

    edge[num].u=u,edge[num].v=v,edge[num++].val=val;

}

LL directed_MST(int root,int n,int e)

{

    int i,j,u,v,cnt=0;

    LL ans=0;

    while(1)

    {

        for(i=1;i<=n;i++) in[i]=inf;

        for(i=0;i<e;i++)//找最小入边

        {

            u=edge[i].u;

            v=edge[i].v;

            if(in[v]>edge[i].val&&v!=u)

            {

                in[v]=edge[i].val;

                pre[v]=u;

            }

        }

        for(i=1;i<=n;i++)

        {

            if(i==root) continue;

            if(in[i]==inf) return -1;

        }

        memset(vi,-1,sizeof(vi));

        memset(id,-1,sizeof(id));

        cnt=0;

        in[root]=0;

        for(i=1;i<=n;i++)//枚举每个可能是环上的点

        {

            ans+=in[i];

            v=i;

            while(vi[v]!=i&&id[v]==-1&&v!=root)

            {

                vi[v]=i;v=pre[v];

            }

            if(v!=root&&id[v]==-1)//如果退出上个循环的条件不是v==root，就是vi[i]==i，即找到了环

            {

                id[v]=++cnt;

                for(u=pre[v];u!=v;u=pre[u])//对环进行统一标号

                    id[u]=cnt;

            }

        }

        if(cnt==0) break;//无环 ，退出

        for(i=1;i<=n;i++) if(id[i]==-1) id[i]=++cnt;

        for(i=0;i<e;i++)//进行缩点，并修改每条边的权值。

        {

            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].val-=in[v];

        }

        n=cnt;

        root=id[root];

    }

    return ans;

}

int main()

{

    int i,j,n,m,a,b;

    while(scanf("%d%d",&n,&m)!=EOF)

    {

        num=0;

        for(i=1;i<=n;i++)

            scanf("%lf%lf",&p[i].x,&p[i].y);

        for(i=1;i<=m;i++)

        {

            scanf("%d%d",&a,&b);

            if(a!=b)

            add(a,b,Dis(p[a],p[b]));

        }

        double ans=directed_MST(1,n,num);

        if(ans==-1)

            printf("poor snoopy\n");

        else

            printf("%.2lf\n",ans);

    }

    return 0;

}