3366 【模板】最小生成树（Kruskal）

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出orz

输入输出格式

输入格式：

第一行包含两个整数N、M，表示该图共有N个结点和M条无向边。（N<=5000，M<=200000）

接下来M行每行包含三个整数Xi、Yi、Zi，表示有一条长度为Zi的无向边连接结点Xi、Yi

输出格式：

输出包含一个数，即最小生成树的各边的长度之和；如果该图不连通则输出orz

代码：

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
struct edge{
    int s, e, w;
    edge(int ss, int ee, int ww):s(ss), e(ee), w(ww){}
    bool operator<(const edge&e1){
        return w<e1.w;
    }
};
vector<edge>edges;
vector<int>par;
int getpar(int x){
    if(par[x]!=x){
        par[x] = getpar(par[x]);
    }
    return par[x];
}
void merge(int x, int y){
    if(getpar(x)==getpar(y)) return;
    par[getpar(x)] = getpar(y);
}
int main(){
    int n, m;
    int i;
    cin>>n>>m;
    par.resize(n+1);
    for(i = 1; i <= n; i++)
        par[i] = i;
    for(i = 1; i <= m; i++){
        int s, e, w;
        cin>>s>>e>>w;
        edges.push_back(edge(s,e,w));
    }
    sort(edges.begin(),edges.end());
    int done = 0, way = 0;
    for(i = 0; i <edges.size(); i++){
        if(getpar(edges[i].s)!=getpar(edges[i].e)){
            merge(edges[i].s, edges[i].e);
            done++;
            way += edges[i].w;
        }
        if(done == n-1) break;
    }
    cout<<way<<endl;
    return 0;
}

备注：

Frankly speaking，我觉得Kruskal比Prim要好理解+好写。只是把边排序，再加一个并查集就可以实现。在某种程度上更好理解。

再去看看图论的内容。