Constructing Roads

北大2421

题意：输入一个数n,代表村庄数，然后输入n行n列的数组，第i行第j列代表第i个村庄与第j个村庄的距离。然后输入一个数m,接下来m行，每行代表某两个村庄之间的路已被修；

输出要修的最短距离，保证每两个村庄之间都可以直接或间接的联系。

//2421C++
#include<stdio.h>
#include<string.h>
#include<stdlib.h>

#define Max 1005
int n,a[Max][Max];
int set[Max],f[Max][Max];
struct PP
{
    int d;
    int u,v;
}b[Max];
int cmp(const void *a,const void *b)
{
    struct PP *x,*y;
    x=(struct PP *)a;
    y=(struct PP *)b;
    return x->d>y->d?1:-1;
}

int find(int x)//查找X并压缩路径
{
    int r,i;
    r=x;
    while(r!=set[r])
    {
        r=set[r];
    }
    while(x!=r)
    {
        i=set[x];
        set[x]=r;
        x=i;
    }
    return r;
}

int main()
{
    int m,i,j,k,x,y,f1,f2;
    while(scanf("%d",&n)!=-1)
    {
        memset(f,0,sizeof(f));
        memset(a,0,sizeof(a));
        for(i=1;i<=n;i++)
            set[i]=i;
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=n;j++)
            {
                scanf("%d",&a[i][j]);
            }
        }
        scanf("%d",&m);
        for(i=1;i<=m;i++)
        {
            scanf("%d%d",&x,&y);
            f[x][y]=1;f[y][x]=1;//将已修的路在二维空间里标记为1；
            f1=find(x);
            f2=find(y);
            if(f1!=f2)
                set[f1]=f2;//判断两个根结点，并做处理
        }
        k=0;
        for(i=1;i<n;i++)
        {
            for(j=i+1;j<=n;j++)
            {
                b[k].d=a[i][j];//用结构体存储两村庄间的距离
                b[k].u=i;//记下两村庄的编号
                b[k].v=j;
                k++;
            }
        }
        qsort(b,k,sizeof(b[0]),cmp);//对村庄间的距离进行从小到大排序
        int totle=0;int p1,p2;
        for(i=0;i<k;i++)
        {
            p1=b[i].u;
            p2=b[i].v;
            if(f[p1][p2]!=1)//判断这条路是否已修
            {
                f1=find(p1);
                f2=find(p2);
                if(f1!=f2)//两村庄间的路未修并且不能通过其他方式到达，则加在totle上；
                {
                    set[f1]=f2;
                    totle+=b[i].d;
                }
            }
        }
        printf("%d\n",totle);
    }
    return 0;
    
}