ZJNU 1528 - War--高级

类似于1213取水

可以把空投当作第0个城市

最后将0~n的所有城市跑最小生成树

#include<iostream>
#include<algorithm>
using namespace std;
struct road{
    int from,to,cost;
    bool operator < (const road &a) const{
        return cost<a.cost;
    }
}ar[];
int gp[];
int find(int a){
    return a==gp[a]?a:gp[a]=find(gp[a]);
}
void merge(int a,int b){
    gp[find(b)]=find(a);
}
int main(){
    ios::sync_with_stdio();cin.tie();
    int T,i,n,m,k;
    long long res;
    cin>>T;
    while(T--){
        cin>>n>>m;
        for(i=;i<n;i++){
            ar[i*].from=i+;
            ar[i*].to=;
            cin>>ar[i*].cost;
            ar[i*+].from=;
            ar[i*+].to=i+;
            ar[i*+].cost=ar[i*].cost;
        }
        for(i=n;i<n+m;i++){
            cin>>ar[i*].from>>ar[i*].to>>ar[i*].cost;
            ar[i*+].from=ar[i*].to;
            ar[i*+].to=ar[i*].from;
            ar[i*+].cost=ar[i*].cost;
        }
        sort(ar,ar+(m+n)*);
        for(i=;i<=n;i++)
            gp[i]=i;
        for(k=res=i=;i<(m+n)*;i++){
            if(k==n)
                break;
            if(find(ar[i].from)!=find(ar[i].to)){
                merge(ar[i].from,ar[i].to);
                res+=ar[i].cost;
                k++;
            }
        }
        cout<<res<<'\n';
    }

    return ;
}