Constructing Roads

poj2421:http://poj.org/problem?id=2421

题意：这一题也是最小生成树，不过它要求修的路在下面还有一行是已经修的路所以将其权值赋值为零。题目给的数值的意思是每个点离其他点的距离

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cmath>
using namespace std;
int n,q,num,pa[102],cnt,sum;
int g[101][101],a,b;
struct Node{
    int u;
    int v;
    int w;
    bool operator<(const Node &a)const{
        return w<a.w;
    }
    
}edge[5000];
void UFset(){
    for(int i=1;i<=n;i++)
     pa[i]=-1;
}
int Find(int x){
    int s;
    for(s=x;pa[s]>=0;s=pa[s]);
    while(s!=x){
        int temp=pa[x];
        pa[x]=s;
        x=temp;
    }
    return s;
}
void Union(int R1,int R2){
    int r1=Find(R1);
    int r2=Find(R2);
    int temp=pa[r1]+pa[r2];
    if(pa[r1]>pa[r2]){
      pa[r1]=r2;
      pa[r2]=temp;    
    }
    else{
        pa[r2]=r1;
        pa[r1]=temp;
    }
}
void kruska(){
    UFset();
    num=0;sum=0;
    for(int i=1;i<cnt;i++){
        int u=edge[i].u;
        int v=edge[i].v;
        if(Find(u)!=Find(v)){
            num++;
            sum+=edge[i].w;
            Union(u,v);
        }
        if(num>=n-1)break;
    }
    printf("%d\n",sum);
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
      for(int j=1;j<=n;j++)
        scanf("%d",&g[i][j]);
        scanf("%d",&q);
    for(int i=1;i<=q;i++){
        scanf("%d%d",&a,&b);
        g[a][b]=g[b][a]=0;
    }
    cnt=1;
     for(int i=1;i<=n;i++)
       for(int j=i+1;j<=n;j++){
           edge[cnt].u=i;edge[cnt].v=j;
           edge[cnt++].w=g[i][j];
       }
     sort(edge+1,edge+cnt); 
     kruska();     

}