Kruskal算法

Kruskal算法

紫书P356

这是最小生成树（也叫MST）的算法。
这里的这个排序cmp函数还是比较牛逼的，他是根据w的值排序得到w的下标递增值。
i.e.这个排序可以保留w数组不动，但又可以得到拍好序的下标值，存在r数组中。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
#define cle(a,val) memset(a,(val),sizeof(a))
#define SI(N) scanf("%d",&(N))
#define SII(N,M) scanf("%d %d",&(N),&(M))
#define SIII(N,M,K) scanf("%d %d %d",&(N),&(M),&(K))
#define rep(i,b) for(int i=0;i<(b);i++)
#define rez(i,a,b) for(int i=(a);i<=(b);i++)
#define red(i,a,b) for(int i=(a);i>=(b);i--)
const ll LINF = 0x3f3f3f3f3f3f3f3f;
#define PU(x) puts(#x);
#define PI(A) cout<<(A)<<endl;
#define DG(x) cout<<#x<<"="<<(x)<<endl;
#define DGG(x,y) cout<<#x<<"="<<(x)<<" "<<#y<<"="<<(y)<<endl;
#define DGGG(x,y,z) cout<<#x<<"="<<(x)<<" "<<#y<<"="<<(y)<<" "<<#z<<"="<<(z)<<endl;
#define PIar(a,n) rep(i,n)cout<<a[i]<<" ";cout<<endl;
#define PIarr(a,n,m) rep(aa,n){rep(bb, m)cout<<a[aa][bb]<<" ";cout<<endl;}
const double EPS = 1e-9 ;
/*  ////////////////////////   C o d i n g  S p a c e   ////////////////////////  */
const int MAXN = 1000 + 9 ;
//第i条边的两个端点u v和权值w 和比较的辅助函数r
int u[MAXN],v[MAXN],w[MAXN],r[MAXN];
//并查集p
int p[MAXN];
int n,m;
int cmp(const int i,const int j){ return w[i]<w[j];}
int find(int x){ return p[x]==x?x:p[x]=find(p[x]);}
int Kruskal(){ 
    int ans=0;
    for (int i=0;i<n;i++) p[i]=i;
    for (int i=0;i<m;i++) r[i]=i;
    PIar(r,m)
    sort(r,r+m,cmp);
    PIar(r,m)
    for (int i=0;i<m;i++){
        int e=r[i];
        int x=find(u[e]);
        int y=find(v[e]);
        if (x!=y){
            ans+=w[e];
            p[x]=y;
        }
    }
    return ans;
}

void Solve()
{
    while(cin>>n>>m){
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&u[i],&v[i],&w[i]);
        }
        printf("%d\n",Kruskal());
    }
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("1.in", "r", stdin);
    freopen("1.out","w",stdout);
#endif
//iostream::sync_with_stdio(false);
//cin.tie(0), cout.tie(0);
    // int T;cin>>T;while(T--)
    Solve();
    return 0;
}

以下是测试的数据：
输入
5 6
1 5 2
1 3 2
5 3 6
4 3 1
4 2 3
2 3 4
输出
0 1 2 3 4 5
3 0 1 4 5 2
8