重庆育才集训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;
} 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值