pipioj 1454: 网络工程师（kruskal）

#define bug(x) cout<<#x<<" is "<<x<<endl
#define IO std::ios::sync_with_stdio(0)
#include <bits/stdc++.h>
using namespace  std;
typedef long long ll;
#define mk make_pair
#define pb push_back
const int inf=2147483647;
const int N=1e3+10;
int fa[N],a[N];
vector<int>g[N],v[N];
int find(int x){
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
int n,m,k;
struct node{
    int x,y,w;
}b[100005];


bool cmp(node p,node q){
    return p.w<q.w;
}
int main(){
    IO;
    while(cin>>n>>m>>k){
        for(int i=1;i<=n;i++)fa[i]=i;
        for(int i=1;i<=n;i++){
            cin>>a[i];
        }
        for(int i=1;i<=m;i++){
            int x,y;
            cin>>x>>y;
            b[i].x=x;
            b[i].y=y;
            if(a[x]==a[y]){
                b[i].w=0;
            }
            else{
                b[i].w=1;
            }
        }
        sort(b+1,b+1+m,cmp);
        int ans=0,cnt=0;
        for(int i=1;i<=m;i++){
            //bug(i);
            int x=b[i].x;
            int y=b[i].y;
            x=find(x);
            y=find(y);
            if(x!=y){
                ans+=b[i].w;
                fa[x]=y;
                cnt++;
            }
            if(cnt==n-1)break;
        }
        cout<<ans<<endl;
    }

}
/*
4 4 4
1 2 3 4
1 2
2 3
3 4
4 1

4 3 2
1 1 2 2
1 3
2 3
3 4
*/