【最小生成树+kruskal】杭电 hdu 1863 畅通工程

/* THE PROGRAM IS MADE BY PYY */
/*----------------------------------------------------------------------------//
    Copyright (c) 2012 panyanyany All rights reserved.

    URL   : http://acm.hdu.edu.cn/showproblem.php?pid=1863
    Name  : 1863 畅通工程

    Date  : Monday, February 6, 2012
    Time Stage : half an hour

    Result: 
5320429	2012-02-06 10:00:29	Accepted	1863
0MS	188K	2039 B
C++	pyy


Test Data :

Review :

//----------------------------------------------------------------------------*/

#include <cstdio>
#include <stdlib.h>
#include <string.h>

#include <algorithm>
using namespace std ;

#define MEM(a, v)		memset (a, v, sizeof (a))	// a for address, v for value
#define max(x, y)		((x) > (y) ? (x) : (y))
#define min(x, y)		((x) < (y) ? (x) : (y))

#define INF		(0x3f3f3f3f)
#define MAXN	(103)
#define MAXE	(MAXN*MAXN)

#define DEBUG	/##/

struct EDGE {
	int u, v, w ;
	bool operator< (const EDGE &e) {
		return w < e.w ;
	}
};

int		n, m ;
int		set[MAXN] ;
int		map[MAXN][MAXN] ;

EDGE	edge[MAXE] ;

int find (int x)
{
	if (x == set[x])
		return x ;
	return set[x] = find(set[x]) ;	// 压缩路径
}

inline void merge (int x, int y)
{
	set[x] = set[y] ;
}

int kruskal ()
{
	int i, sum, edgeCnt ;

	// 将边排序
	sort (edge, edge+n) ;
	
	// 并查集初始化，每个点自成集合，根结点为自己
	for (i = 1 ; i <= m ; ++i)
		set[i] = i ;

	sum = edgeCnt = 0 ;
	for (i = 0 ; i < n ; ++i)
	{
		// 查找各集合的根结点
		int x = find (edge[i].u) ;
		int y = find (edge[i].v) ;

		// 根结点不同，则所在集合不同
		if (x != y)
		{
			// 合并两个不同的集合
			merge (x, y) ;
			sum += edge[i].w ;
			// 所有集合中的边数
			++edgeCnt ;
			// 若边数等于 m-1，表示所有点都在一个集合中了
			if (m-1 == edgeCnt)
				break ;
		}
	}

	// 边数小于m-1，表示有两个以上集合不能合并
	if (m-1 > edgeCnt)
		return INF ;
	return sum ;
}

int main ()
{
	int i ;
	int ans ;
	while (scanf ("%d%d", &n, &m), n)
	{
		for (i = 0 ; i < n ; ++i)
			scanf ("%d%d%d", &edge[i].u, &edge[i].v, &edge[i].w) ;
		ans = kruskal () ;
		if (INF == ans)
			puts ("?") ;
		else
			printf ("%d\n", ans) ;
	}
	return 0 ;
}