POJ 1679 次小生成树

判断最小生成树是否唯一，方法是这样的：

1. 对图中每个点，扫描其他的边，如果存在其他权值相同的边，则对改边作标记

2. 然后用 Kruskal或者Prim 求MST

3. 求的MST后，如果该MST不包含作了标记的边，即可判定MST唯一；如果包含作了标记的边，则依次去掉这些边在求MST，如果求的MST权值和原MST权值一样，即可判定MST不唯一。

这里在在第二次计算生成树时，不能每次都将边的used 置为1，只有在第一次可以，因为第一次计算的结果就是最终的结果。而第二次计算的时候是枚举的，会有很多种情况，而used本身的作用就是在第二次生成树算法中和 uni 一起判断使用的，这个只依赖第一次的结果。

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

using namespace std;

#define ELEN 10000

typedef struct Edge {
	int u, v;
	int w;
	int used;
	int uni;
	int del;
}Edge;
Edge e[ELEN];
int p[ELEN];
bool first;

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

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

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

int kruskal(int n, int m) {
	int sum = 0;
	int num = 1;
	for(int i = 0; i < m; i++) {
		if(1 == e[i].del) continue;
		int t1 = find(e[i].u);
		int t2 = find(e[i].v);
		if(t1 != t2) {
			p[t1] = t2;
			num++;
			sum += e[i].w;
			if(first) {
				e[i].used = 1;
			}
		}
	}
	if(num != n) return -1;
	return sum;
}

int main() {
	int t;
	scanf("%d", &t);
	while(t--) {
		int i, j;
		int n, m;
		scanf("%d%d", &n, &m);
		first = true;
		init(n);
		for(i = 0; i < m; i++) {
			scanf("%d%d%d", &e[i].u, &e[i].v, &e[i].w);
			e[i].used = 0;
			e[i].uni = 0;
			e[i].del = 0;
		}
		sort(e, e+m, cmp);
		for(i = 0; i < m; i++) {
			for(j = 0; j < m; j++) {
				if(i == j) continue;
				if(e[i].w == e[j].w) e[i].uni = 1;
			}
		}
		int ans = kruskal(n, m);
		for(i = 0; i < m; i++) {
			if(1 == e[i].used && 1 == e[i].uni) break;
		}
		if(i >= m) {
			printf("%d\n", ans);
			continue;
		}
		else {
			first = false;
			for(i = 0; i < m; i++) {
				if(1 == e[i].used && 1 == e[i].uni) {
					init(n);
					e[i].del = 1;
					if(ans == kruskal(n, m)) {
						printf("Not Unique!\n");
						break;
					}
					e[i].del = 0;
				}
			}
			if(i >= m) {
				printf("%d\n", ans);
			}
		}
	}
	return 0;
}