hihoCoder#1098 最小生成树二·Kruscal算法

原题地址

 

以前没写过Kruscal算法，写了才知道原来比Prime算法简单多了。。。

并查集的应用太经典了！

 

代码：

 1 #include <iostream>

 2 #include <cstdlib>

 3 

 4 using namespace std;

 5 

 6 #define MAX_EDGE 1000008

 7 #define MAX_POINT 100008

 8 

 9 struct Edge {

10   int a;

11   int b;

12   int len;

13 };

14 

15 int mycmp(const void *a, const void *b) {

16   Edge *pa = (Edge *) a;

17   Edge *pb = (Edge *) b;

18   return pa->len - pb->len;

19 }

20 

21 int N, M;

22 Edge e[MAX_EDGE];

23 int p[MAX_POINT];

24 

25 int find(int i) {

26   if (p[i] == i)

27     return i;

28   return p[i] = find(p[i]);

29 }

30 

31 void merge(int i, int j) {

32   int pi = find(i);

33   int pj = find(j);

34   p[pi] = pj;

35 }

36 

37 int kruscal() {

38   int res = 0;

39 

40   qsort(e, M, sizeof(Edge), mycmp);

41   for (int i = 0; i < M; i++) {

42     if (find(e[i].a) == find(e[i].b))

43       continue;

44     res += e[i].len;

45     merge(e[i].a, e[i].b);

46   }

47 

48   return res;

49 }

50 

51 int main() {

52   scanf("%d%d", &N, &M);

53   for (int i = 1; i <= N; i++)

54     p[i] = i;

55   for (int i = 0; i < M; i++)

56     scanf("%d%d%d", &(e[i].a), &(e[i].b), &(e[i].len));

57 

58   printf("%d\n", kruscal());

59   return 0;

60 }