hdu1102

http://acm.hdu.edu.cn/showproblem.php?pid=1102

最小生成树(模板题)

3

0 990 692

990 0 179

692 179 0

1

1 2

一共3个村子，下面是第i行j列 i村子和j村子之间建路需要的距离

下面是一个k

代表有k条路已经修好了，1村子和2村子之间以修路

 #include<iostream>
 #include<math.h>
 #include<string.h>
 #include<stdlib.h>
 #include<stdio.h>
 using namespace std;
 const int N=;
 struct stu{
     int u;
     int v;
     int w;
 }p[N];
 int q[][];
 int father[N];
 int n,m;
 int cmp(const void *a,const void *b)
 {
     return (*(struct stu*)a).w > (*(struct stu*)b).w ?:-;
 }
 int find(int x)
 {
     if(father[x]!=x)
     father[x]=find(father[x]);
     return father[x];
 }
 int make(int a,int b)
 {
     int h=;
     int f1=find(a);
     int f2=find(b);
     if(f1>f2)
     {
         father[f1]=f2;
         h=;
     }
     else if(f1<f2)
     {
         father[f2]=f1;
         h=;
     }
     return h;
 }
 int kruskal()
 {
     int cnt=;
     int s=;
     for(int i=;i<m;i++)
     {
         if(make(p[i].u,p[i].v))
         {
             cnt++;
             s+=p[i].w;
         }
         if(cnt==n-)
         return s;
     }
     return s;
 }

 int main()
 {
     //freopen("in.txt","r",stdin);
     int k;
     while(~scanf("%d",&n))
     {
         for(int i=;i<=n;i++)
         father[i]=i;
         for(int i=;i<=n;i++)
         {
             for(int j=;j<=n;j++)
             {
                 scanf("%d",&q[i][j]);
             }
         }
         scanf("%d",&k);
         int a,b;
         while(k--)
         {
             scanf("%d%d",&a,&b);
             q[a][b]=;//已经建路，清0
         }
         m=;
         for(int i=;i<=n;i++)
         {
             for(int j=i+;j<=n;j++)
             {
                 p[m].u=i;
                 p[m].v=j;
                 p[m++].w=q[i][j];
             }
         }
         qsort(p,m,sizeof(struct stu),cmp);
 //        for(int i=0;i<m;i++)
 //        {
 //            printf("%d %d %d\n",p[i].u,p[i].v,p[i].w);
 //        }
         printf("%d\n",kruskal());

     }
     return ;
 }