pta7-20 畅通工程之局部最小花费问题（Kruskal算法）

题目链接：https://pintia.cn/problem-sets/15/problems/897

题意：给出n个城镇，然后给出n×（n-1）/2条边，即每两个城镇之间的边，包含起始点，终点，修建的花费，以及是否已经修建。需要求出最小花费让城镇两两可达。

思路：最小生成树变形题，首先将边排序，将已经修好的放在前面，未修好的放在后面，并均按照花费升序排列。之后用并查集将已经连通的城镇并在一起，然后对未修好的道路使用kruskal算法，直到所有的城镇两两可达，此时的花费即最小。

AC代码：

 #include<bits/stdc++.h>
 using namespace std;

 struct node{
     int s,e,c,f;
 }a[];

 bool cmp(node x,node y){
     if(x.f==y.f) return x.c<y.c;
     return x.f>y.f;
 }

 int n,m,res;
 int root[];

 int getr(int k){
     if(root[k]==k) return k;
     else return root[k]=getr(root[k]);
 }

 int main(){
     scanf("%d",&n);
     m=n*(n-)/;
     for(int i=;i<m;++i)
         scanf("%d%d%d%d",&a[i].s,&a[i].e,&a[i].c,&a[i].f);
     sort(a,a+m,cmp);
     for(int i=;i<=n;++i)
         root[i]=i;
     for(int i=;i<m;++i){
         int rs=getr(a[i].s),re=getr(a[i].e);
         if(a[i].f==)
             root[re]=rs;
         else if(rs!=re)
             res+=a[i].c,root[re]=rs;
     }
     printf("%d\n",res);
 }