hdu1102

第一道最小生成树问题

#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdio.h>
#define maxn 10000+100
using namespace std;
int r[maxn];
int u[maxn];
int v[maxn];
int w[maxn];
int p[maxn];
int map[110][110];
bool inmap[110][110];
int cmp(const int i,const int j){return w[i]<w[j];}
int find(int x)
{
    return p[x]==x?x:p[x]=find(p[x]);
}
void merge(int x,int y)
{
    int a=find(x),b=find(y);
    p[a]=b;
}
int main()
{
    int cnt;
    int n,q;
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
            scanf("%d",&map[i][j]);
        memset(inmap,0,sizeof(inmap));
        scanf("%d",&q);
        for(int i=1;i<=n;i++)
        {
            p[i]=i;
        }
        for(int i=1;i<=q;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            merge(u,v);
            inmap[u][v]=inmap[v][u]=true;
        }
        int cnt=0;
        for(int i=1;i<=n;i++)
            for(int j=1;j<i;j++)//读取一半的图即可
                if(!inmap[i][j])
                {
                    u[cnt]=i;
                    v[cnt]=j;
                    w[cnt++]=map[i][j];
                }
        int ans=0;
        for(int i=0;i<cnt;i++) r[i]=i;
        //cout<<"cnt="<<cnt<<endl;
        sort(r,r+cnt,cmp);
        for(int i=0;i<cnt;i++)
        {
            int e=r[i];int x=find(u[e]);int y=find(v[e]);
            if (x!=y)
                {ans+=w[e];merge(x,y);}
        }
        printf("%d
",ans);

    }
    return 0;
}

题目是说：n个城市间修建道路使之联通，其中有些路已经修好，再添加几条道路，是城市间能联通，且路程最短......