重庆育才集训2018【夏】最小生成树

问题描述

已知一个包含 n
个点 m 条边的森林，第 i 个点的权值为 ai。在图中点 i 和点 j 之间连接一条边的代价为 ai+aj，并且每个点只可以添加一条边，求将该森林变为一棵树的最小代价。如果不存在可行方案，输出 −1。
输入格式

第一行包含两个整数 n,m。 第二行包含 n 个整数，表示每个点的权值。 接下来 m 行每行包含两个整数 u,v(1≤u,v≤n)，表示点 u 和点 v之间存在一条边相连。
输出格式

输出一个整数。
样例输入
5 2
1 2 5 3 4
1 3
2 4
样例输出
10
数据规模和约定
对于 30%
数据：所有点的权值都相同。
对于 100%
数据：0≤m≤n≤105，0≤ai≤109，输入保证 n 个点 m条边组成一个森林（可能是一颗完整树）。
时空限制
1S/512MB

思路
求出已有连通块个数k，[连通这k块需要k-1条边，每条边需要2个点]，现需要求的这些点的权值和最小。[每个连通块上至少要去一个点，不然没法将其合并进其他连通块]
实现
并查集 + 优先队列
代码

#include <bits/stdc++.h>
using namespace std;
const long long maxn=111111;
long long n,m;
long long fa[maxn];
long long a[maxn];
bool vis[maxn];
void init(){
    for(long long i=0;i<=n+19;i++){
        fa[i]=i;
    }
}
long long get(long long x) {
    if (fa[x] == x) {
        return x;
    }
    return fa[x]=get(fa[x]);
}
void merge(long long x,long long y){
    x=get(x);
    y=get(y);
    if(x!=y){
        fa[y]=x;
    }
}
long long ans=0;
vector<long long> point; 
priority_queue<long long,vector<long long>, greater<long long> > q[maxn];
priority_queue<long long,vector<long long>, greater<long long> > all;
int main(){
    cin>>n>>m;
    init();
    bool is=1;
    for(long long i=1;i<=n;i++){
        cin>>a[i];
        if(a[i]!=a[1]){
            is=0;
        } 
    }
    for(long long i=0;i<m;i++){
        long long u,v;
        cin>>u>>v;
        if(u==v || fa[u]==fa[v]){
            cout<<-1;
            return 0;
        }
        merge(u,v);
    }
    long long num=0;
    for(long long i=1;i<=n;i++){
        if(fa[i]==i){
            num++;
            point.push_back(i);
        }       
    }
    if(2*num-2>n){
        cout<<-1;
        return 0;
    }
    if(is){
        cout<<(2*num-2)*a[1];
        return 0;
    }
    for(long long i=1;i<=n;i++){
        q[get(i)].push(a[i]); 
    }
    for(long long i=0;i<point.size();i++){
        long long u=point[i];
        ans+=q[u].top();
        q[u].pop();
        while(!q[u].empty()){
            long long tmp=q[u].top();
            q[u].pop();
            all.push(tmp);
        }
    }
    long long c=num-2;
    while(c--){
        long long tmp=all.top();
        ans+=tmp;
        all.pop();
    }
    cout<<ans<<endl;

    return 0;
}