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[10000];

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[105];

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-1)/2;
    for(int i=0;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=1;i<=n;++i)
        root[i]=i;
    for(int i=0;i<m;++i){
        int rs=getr(a[i].s),re=getr(a[i].e);
        if(a[i].f==1)
            root[re]=rs;
        else if(rs!=re)
            res+=a[i].c,root[re]=rs;    
    }
    printf("%d
",res);
}