ZOJ 1542 Kruskal

一开始输出弄错了，卡了好久

直接上代码了

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

#define ELEN 15005
#define VLEN 1005

typedef struct Edge {
	int u, v;
	int w;
}Edge;
Edge e[ELEN];
int p[VLEN];
int aans[VLEN];

bool cmp(Edge a, Edge b) { 
	return a.w < b.w;
}

int find(int x) {
	return p[x] == x ? x : (p[x] = find(p[x]));
}

void init(int n) {
	for(int i = 0; i <= n; i++) {
		p[i] = i;
	}
}

int kruskal(int n, int m) {
	int index = 0;
	int mmax = 0;
	memset(aans, 0, sizeof(aans));
	for(int i = 0; i < m; i++) {
		int t1 = find(e[i].u);
		int t2 = find(e[i].v);
		if(t1 != t2) {
			p[t1] = t2;
			aans[index++] = i;
			if(mmax < e[i].w) {
				mmax = e[i].w;
			}
		}
		if(index >= n-1) break;
	}
	printf("%d\n%d\n", mmax, index);
	for(int i = 0; i < index; i++) {
		printf("%d %d\n", e[aans[i]].u, e[aans[i]].v);
	}
}

int main() {
	int n, m;
	while(scanf("%d%d", &n, &m) != EOF) {
		memset(e, 0, sizeof(e));
		for(int i = 0; i < m; i++) {
			scanf("%d%d%d", &e[i].u, &e[i].v, &e[i].w);
		}
		init(n);
		sort(e, e+m, cmp);
		kruskal(n, m);
	}
	return 0;
}