B3604 [图论与代数结构 302] 最短树问题_2

题目描述
在离散数学课程的学习中，大家学习了最短树相关的知识，现在来检查一下大家的学习情况。

给定一张 nn 个点 mm 条边的带权连通无向图，点的编号从 11 到 nn，可能存在重边和自环。现在，请你求出这张图总边权和最小的支撑树（最短树）的边权之和。即，找到一颗支撑树，满足选用边的边权和最小，输出这个边权和。

输入格式
第一行两个正整数 n, mn,m ，分别表示给定无向图的点数和边数。

接下来 mm 行，每行三个非负整数 u, v, wu,v,w 表示一条连接 uu 和 vv 这两个点的无向边和这条边的边权 ww。

输出格式
输出仅一行一个非负整数，表示最短树的边权之和。

输入输出样例
输入 #1复制
4 5
1 3 1
1 2 2
2 3 0
3 4 3
1 4 8
输出 #1复制
4
说明/提示
对于所有数据，满足 1\leq n \leq 1000001≤n≤100000，1 \leq m \leq 3000001≤m≤300000，边权均是 [0, {10}^9][0,10
9
] 中的整数，保证输入的图合法且满足题目中的限制条件。

请注意答案上界的大小，可能需要使用 C++ 中的 long long int 类型。

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long LL;
const int N=1e5+10,M=2e5+10;
int n,m;
int p[N];

int find(int x)
{
    if(p[x] != x) p[x]=find(p[x]);
    return p[x];
}

struct edge
{
    int a,b,w;
    
    bool operator<(const edge &W)const
    {
         return w<W.w;     
    }
}edges[M];

LL Kru()
{
    sort(edges,edges+m);
    
    for(int i=1;i<=n;i++) p[i]=i;
    
    int res=0;
    for(int i=0;i<m;i++)
    {
         int a=edges[i].a,b=edges[i].b,w=edges[i].w;
         a=find(a),b=find(b);
        
         if(a != b)
         {
             p[a]=b;
             res+=w;
         }
    }
    return res;
}

int main()
{
    cin>>n>>m;

    for(int i=0;i<m;i++)
    {
        int a,b,w;
        scanf("%d%d%d",&a,&b,&w);
        
        edges[i]={a,b,w};
    }
    
    cout<<Kru();
    
    return 0;
}