天天看點

最小生成樹kruskal算法(并查集)

還是圖的算法,最小生成樹。

kruskal算法和dijkstra算法有異曲同工之處,都有貪心的思想在其中。

先讀入邊集,将所有的邊排序(從小到大),依次加入另一個空集(最小生成樹),如果和其中已有的邊形成回路,就跳過,看下一條邊。用的是并查集判重。最後的解即為最小生成樹。

代碼如下:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

int fa[],n,i,j,m,ans;

struct node{
    int sta;
    int fin;
    int data;
}set[];

int find(int x){//并查集操作 
    int r=x;
    while (fa[r]!=r) r=fa[r];//找父親 
    int i=x,t;
    while (i!=r){//路徑壓縮 
        t=fa[i];
        fa[i]=r;
        i=t;
    }
    return r;
}
int main(){
    scanf("%d%d",&n,&m);
    for (i=;i<=m;++i){
        scanf("%d%d%d",&set[i].sta,&set[i].fin,&set[i].data);
    }
    for (i=;i<=m;++i){
        fa[i]=i;
    }//初始化 
    for (i=;i<=m;++i){
        if (find(set[i].sta)!=find(set[i].fin)) ans+=set[i].data;
        fa[find(set[i].fin)]=set[i].sta; //如果父親不一樣(不在同一個集合,即沒有形成回路),就加入生成樹中 
    }
    printf("%d",ans);
}