记录做题,今天的题目是并查集加克鲁斯卡尔算法求最小带权生成树。
题目大意就是说,给你m组顶点,顶点之间互有边为完全图,并且顶点和顶点之间有权值。然后,你要找到他的一个最小生成树。
这里建图选择并查集,原因之一是因为如果用前向星链表,互相建边一定会超时,因此我们只需要,把一组顶点拴在一个链子上代表一个完全图就欧克了。所以建立结构体储存pre和now。
之后用kru算法,选择·最小边如果不在一个集合就加入集合,累加权值,直到所有顶点都加入集合,找到最小生成树,如果发现最小生成树的边不是n-1条边,那说明有的顶点之间不能相互到达,所以无法生成树,输出-1即可。
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define N 400005
#define PLL pair<ll,ll>
ll ans;
ll a;
struct node{
ll u,v,w;
}num[N];
bool cmp(node a,node b){
return a.w<b.w;
}
ll f[N];
int find(int x){
return x==f[x]?x:f[x]=find(f[x]);
}
ll sum=0;
void join(node a){
ll uu=a.u;
ll vv=a.v;
ll ww=a.w;
ll fv=find(vv);
ll fu=find(uu);
if(fv!=fu){
f[fu]=fv;
ans+=ww;
sum++;
}
}
int cnt;
void solve() {
ll n,m,k,c;
ll pre=-1;
cin>>n>>m;
for(int i=1;i<=n;i++)f[i]=i;
for(int i=1;i<=m;i++){
cin>>k>>c;
for(int j=1;j<=k;j++){
cin>>a;
if(j>1){
num[++cnt]={pre,a,c};
}
pre=a;
}
}
sort(num+1,num+1+cnt,cmp);
for(int i=1;i<=cnt;i++){
join(num[i]);
}
if(sum!=n-1)cout<<"-1"<<endl;
else {
cout<<ans<<endl;
}
}
int main() {
solve();
}