PAT 最小生成树的唯一性

题目描述
给定一个带权无向图，如果是连通图，则至少存在一棵最小生成树，有时最小生成树并不唯一。本题就要求你计算最小生成树的总权重，并且判断其是否唯一。
输入格式
首先第一行给出两个整数：无向图中顶点数 N（≤500）和边数 M。随后 M 行，每行给出一条边的两个端点和权重，格式为“顶点1 顶点2 权重”，其中顶点从 1 到N 编号，权重为正整数。题目保证最小生成树的总权重不会超过 230
。
输出格式
如果存在最小生成树，首先在第一行输出其总权重，第二行输出“Yes”，如果此树唯一，否则输出“No”。如果树不存在，则首先在第一行输出“No MST”，第二行输出图的连通集个数。
输入样例
5 7
1 2 6
5 1 1
2 3 4
3 4 3
4 1 7
2 4 2
4 5 5
输出样例
11
Yes

通过并查集以及Kruskal算法进行判断，细节通过代码讲述

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
struct edges
{
	int st;
	int ed;
	int weight;
	bool operator<(const edges& a)const
	{
		return weight < a.weight;
	}
};
// 边集的结构体定义以及运算符重载
int m, n;
vector<int>pre;
vector<edges> e;
int find(int x)
{
	if (x == pre[x])
		return x;
	return find(pre[x]);
}
void unions(int x, int y)
{
	int fx = find(x), fy = find(y);
	if (fx != fy)
	{
		if (fx < fy)
			pre[fy] = fx;
		else pre[fx] = fy;
	}
}
int main()
{
	cin >> m >> n;
	pre.resize(m + 1);
	e.resize(n);
	for (int i = 0; i <= m; i++)
		pre[i] = i;
	for (int i = 0; i < n; i++)
		cin >> e[i].st >> e[i].ed >> e[i].weight;
	sort(e.begin(), e.end());
	bool flag = 0;
	int cnt = 0,sum=0;
	// 边集按照权值从小到大排序
	for (int i = 0; i < n; i++)
	{
		int u = find(e[i].st), v = find(e[i].ed);
		// 取出一个边集，判断是否构成环
		if (u != v)
		{
			// 若没有构成环
			for (int j = i+1; j < n; j++)
			{
				// 从当前边的下一个继续查找
				if (e[j].weight == e[i].weight)
				{
					//如果有 边的权重等于当前权重
					//且起点与终点与我当前边相同
					//该点也可以作为最小生成树的边
					int pu = find(e[j].st), pv = find(e[j].ed);
					if ((pu == u && pv == v) || (pu == v && pv == u))
						flag = 1;
					else break;
				}
			}
			unions(u, v);
			cnt++;
			sum += e[i].weight;
		}
	}
	if (cnt == m - 1)
	{
		if (flag)
		{
			printf("%d\n", sum);
			printf("No");
		}
		else
		{
			printf("%d\n", sum);
			printf("Yes");
		}
	}
	else
	{
		cnt = 0;
		for (int i = 1; i <= m; i++)
		{
			if (pre[i] == i)
				cnt++;
		}
		printf("No MST\n");
		printf("%d", cnt);
	}
	return 0;
}