1143. 联络员【最小生成树】

幽殇默

于 2021-10-14 15:37:20 发布

阅读量203

点赞数

CC 4.0 BY-SA版权

分类专栏：最小生成树文章标签： c++

本文链接：https://blog.youkuaiyun.com/bettle_king/article/details/120765493

最小生成树专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种使用并查集优化的解决方案，针对比赛中的节点连接问题。通过比较和合并操作，提升查找效率，适用于大规模数据集。主要涉及结构化编程、并查集应用及竞赛策略。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述
https://www.acwing.com/problem/content/1145/

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
struct node{int a,b,c;};
bool cmp(node a,node b) {return a.c<b.c;}
vector<node>ve1,ve2;
int p[N],n,m;
int find(int x)
{
    if(x!=p[x]) p[x]=find(p[x]);
    return p[x];
}
int solve()
{
    int res=0;
    for(int i=1;i<=n;i++) p[i]=i;
    for(int i=0;i<ve1.size();i++)
    {
        int a=ve1[i].a,b=ve1[i].b,c=ve1[i].c;
        if(find(a)!=find(b)) p[find(a)]=find(b);
        res+=c;
    }
    sort(ve2.begin(),ve2.end(),cmp);
    for(int i=0;i<ve2.size();i++)
    {
        int a=ve2[i].a,b=ve2[i].b,c=ve2[i].c;
        if(find(a)!=find(b)) p[find(a)]=find(b),res+=c;
    }
    return res;
}
int main(void)
{
    cin>>n>>m;
    for(int i=0;i<m;i++)
    {
        int a,b,c,d; cin>>d>>a>>b>>c;
        if(d==1) ve1.push_back({a,b,c});
        else ve2.push_back({a,b,c});
    }
    cout<<solve()<<endl;
    return 0;
}