Kruskal-并查集-天梯赛L3-最小生成树的唯一性

给定一个带权无向图，如果是连通图，则至少存在一棵最小生成树，有时最小生成树并不唯一。本题就要求你计算最小生成树的总权重，并且判断其是否唯一。

输入格式：

首先第一行给出两个整数：无向图中顶点数 $N$（$\le 500$）和边数 $M$。随后 $M$ 行，每行给出一条边的两个端点和权重，格式为“顶点1 顶点2 权重”，其中顶点从 1 到$N$ 编号，权重为正整数。题目保证最小生成树的总权重不会超过 2302^{30}2​30​​。

输出格式：

如果存在最小生成树，首先在第一行输出其总权重，第二行输出“Yes”，如果此树唯一，否则输出“No”。如果树不存在，则首先在第一行输出“No MST”，第二行输出图的连通集个数。

输入样例 1：

5 7
1 2 6
5 1 1
2 3 4
3 4 3
4 1 7
2 4 2
4 5 5

输出样例 1：

11
Yes

输入样例 2：

4 5
1 2 1
2 3 1
3 4 2
4 1 2
3 1 3

输出样例 2：

4
No

输入样例 3：

5 5
1 2 1
2 3 1
3 4 2
4 1 2
3 1 3

输出样例 3：

No MST
2

作者: 陈越

单位: 浙江大学

时间限制: 400 ms

内存限制: 64 MB

代码长度限制: 16 KB

**思路：** 判断联通性可以先跑一遍dfs，也可以kruskal的同事求解连通性，kruskal跑一遍之后加入的边数cnt,图的节点数n，那么联通分支的数量为n-cnt。 这里我先dfs的，后面判断唯一性就是将之前的最小生成树的某一条边去掉，然后再跑一遍，总共n-1次，看每次返回的结果res是否与第一次算的结果ans相等。相等则不唯一，否则所有的都不相等，则唯一。 不过天梯的数据有点坑，我调了好久才AC，结果把感觉不会错的地方注释掉，居然就AC了，奇怪了。 code speaks louder than words.

#include<bits/stdc++.h>
#define per(i,a,b) for(int i = (a);i <= (b);++i)
#define rep(i,a,b) for(int i = (a);i >= (b);--i)
#define INF 0x3f3f3f3f
typedef long long LL;
using namespace std;
const int maxn = 3e5;
int n = 0,m = 0,cnt = 0;
bool vis[maxn+10];
int f[500+10];
int ran[500+10];
int ans = 0;
bool flag = true; 
set<int> st,st0;
struct node{
	int from,to,w;
};
node edge[maxn+10];
vector<int> g[maxn+10];
void init(){
	per(i,1,n){
		ran[i] = 0;
		f[i] = i;
	}
}
void dfs(int s){
	vis[s] = true;
	int size = g[s].size();
	per(i,0,size-1){
		int v = g[s][i];
		if(!vis[v]){
			dfs(v);
		}
	}
}
int find(int u){
	if(f[u] == u){
		return u;
	}
	return f[u] = find(f[u]);
}
bool same(int a,int b){
	return find(a) == find(b);
}
void unite(int u,int v){
	v = find(v);u = find(u);
	if(v == u){
		return ;
	}else{
		if(ran[u] > ran[v]){
			f[v] = u;
		}else{
			f[u] = v;
			if(ran[u] == ran[v]){
				++ran[v];
			}
		}
	}
}
bool cmp(node a,node b){
	return a.w < b.w;
}
int kruskal(int id){
	init();
	ans = 0;
	int cnt = 0,i = 1;
	while(i <= m){//如果下面不注释掉，这里改成true，那么测试7过了，但是测试3段错误，不知道什么鬼 
		if(cnt == n-1){
			break;
		}
		if(vis[i] || same(edge[i].from,edge[i].to)){
			++i;
			continue;
		}
		ans += edge[i].w;
		unite(edge[i].from,edge[i].to);
		st.insert(i);
		++cnt;
		++i;
	}
	/*
	if(cnt < n-1){//不知道为什么一定要把这个注释掉测试3和测试7才能过，AC 
		return -1;
	}
	*/
	if(id == 0){
		st0 = st;
	}
	return ans;
}
bool kruskal2(int ans0){	
	memset(vis,false,sizeof(vis));
	set<int>::iterator it = st0.begin();
	for(it = it;it != st0.end();++it){
		vis[*it] = true;
		int res = kruskal(1);
		if(res == ans0){
			return false;
		}
		vis[*it] = false;
	}
	return true;
}
int main(){
	while(~scanf("%d %d",&n,&m)){
		st0.clear();
		per(i,1,n){
			g[i].clear();
		}
		memset(vis,false,sizeof(vis));
		per(i,1,m){
			int u = 0,v = 0,w = 0;
			scanf("%d %d %d",&u,&v,&w);
			edge[i].from = u;edge[i].to = v;edge[i].w = w;
			g[u].push_back(v);g[v].push_back(u);
		}
		int num = 0;
		per(i,1,n){
			if(!vis[i]){
				dfs(i);
				++num;
			}
		}
		if(num > 1){
			printf("No MST\n");
			printf("%d\n",num);
			continue;
		}
		sort(edge+1,edge+1+m,cmp);
		memset(vis,false,sizeof(vis));
		int ans0 = kruskal(0);
		printf("%d\n",ans0);
		if(kruskal2(ans0)){
			printf("Yes\n");
		}else{
			printf("No\n");
		}
	}
	
	return 0;
}