還是圖的算法,最小生成樹。
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);
}