hihocoder1098最小生成树（kruscal算法）

kruscal算法描述：

kruscal算法的思路是：最初，把所有节点都看成孤立的集合，将图中所有的边按权重从小到大排序，然后依次遍历这些边，若边的两个端点在两个不同的集合中，则合并这条边的端点所属的两个集合，直到选出n-1条边将图中的所有n个节点都合并到了同一个集合，n-1次合并就选出了n-1条边，由这n-1条边和图上的n哥节点所构成的就是我们需要的该图的最小生成树。

kruscal算法的性能依赖于边的数目，故对于稀疏图的最小生成树问题，采用kruscal算法比较优越。

我的代码：

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

#define MAXN 100005
#define INF 0x7fffffff

struct edge
{
int u, v, w;
edge(int _u, int _v, int _w):u(_u),v(_v),w(_w){}
bool operator<(const edge &x)const
{
return w<x.w;
}
};

vector<edge> e;
int n, m;

struct unionFindSet
{
int st[MAXN];
void init()
{
for(int i=1; i<=n; ++i) st[i] = i;
}
int findSet(int x)
{
if(x==st[x]) return x;
return st[x] = findSet(st[x]);
}
void unionSet(int x, int y)
{
int sx = findSet(x), sy = findSet(y);
st[sx] = sy;
}
}ufs;

int kruscal()
{
sort(e.begin(), e.end());
int ans = 0, cnt = 0;
ufs.init();
for(int i=0; i<e.size()&&cnt<n-1; ++i)
{
int su = ufs.findSet(e[i].u), sv = ufs.findSet(e[i].v);
if(su!=sv)
{
ufs.unionSet(su, sv);
ans += e[i].w;
}
}
return ans;
}

int main()
{
while(cin>>n>>m)
{
e.clear();
while(m--)
{
int u, v, w;
cin>>u>>v>>w;
e.push_back(edge(u, v, w));
}
cout<<kruscal()<<endl;
}
return 0;
}

题目链接：http://hihocoder.com/problemset/problem/1098