poj 1751 Highways 最小生成树 prim

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<iostream>
using namespace std;
const int inf=0x3f3f3f3f;
const int N=1024;
bool vis[N];
int dis[N];
int e[N][N],a[N],b[N],cnt,use[N];
int u1[N],v1[N];

void prim(int n)
{
    int i,j,u,v,w;
    cnt=0;
    for(i=1; i<=n; i++)
    {
        use[i]=1;
        dis[i]=e[1][i];
    }
    memset(vis,0,sizeof(vis));
    vis[1]=1;
    for(i=2; i<=n; i++)
    {
        w=inf;
        u=-1;
        for(j=1; j<=n; j++) if(!vis[j]&&w>dis[j]) w=dis[j],u=j;
        vis[u]=1;
        u1[cnt]=use[u];
        v1[cnt]=u;
        cnt++;
        for(v=1; v<=n; v++) if(!vis[v])
                if(dis[v]>e[u][v])
                {
                    dis[v]=e[u][v];
                    use[v]=u;
                }
    }
}

int main()
{
    int n,m,i,j,u,v,t;
    while(~scanf("%d",&n))
    {
        for(i=1; i<=n; i++)
            scanf("%d%d",&a[i],&b[i]);
        memset(e,0,sizeof(e));
        for(i=1; i<=n; i++)
            for(j=i+1; j<=n; j++)
            {
                t=(a[i]-a[j])*(a[i]-a[j])+(b[i]-b[j])*(b[i]-b[j]);
                e[i][j]=e[j][i]=t;
            }
        scanf("%d",&m);
        for(i=1; i<=m; i++)
        {
            scanf("%d%d",&u,&v);
            e[u][v]=e[v][u]=0;
        }
        prim(n);
        int flag=0;

        for(i=0; i<cnt; i++)
        {
            if(e[u1[i]][v1[i]]==0) continue;
                    flag=1;
                    printf("%d %d\n",u1[i],v1[i]);
        }
    if(flag==0) printf("\n");
    }
    return 0;
}