问题描述
已知一个包含 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;
}