7-1 最小生成树的唯一性 (35 分)

7-1 最小生成树的唯一性 (35 分)

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

输入格式：

首先第一行给出两个整数：无向图中顶点数 N（≤500）和边数 M。随后 M 行，每行给出一条边的两个端点和权重，格式为“顶点1 顶点2 权重”，其中顶点从 1 到N 编号，权重为正整数。题目保证最小生成树的总权重不会超过 2e​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

AC

#include<bits/stdc++.h>
using namespace std;

struct stu {
	int v1,v2;
	int w;
};
int a[501];
vector<stu> v;//

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

int find(int x) {
	if(a[x]==-1) return x;
	else {
		a[x]=find(a[x]);
		return a[x];
	}
}

void merge(int a1,int b) {
	int x=find(a1);
	int y=find(b);
	a[x]=y;
}

int main() {
	int n,m,f=0;
	cin>>n>>m;
	for(int i=0; i<m; i++) {
		stu x;
		cin>>x.v1>>x.v2>>x.w;
		v.push_back(x);
	}

	sort(v.begin(),v.end(),cmp);
	memset(a,-1,sizeof(a));
	int sum=0;
	for(int i=0,num=0; i<m&&num<n-1; i++) {
		int v1=find(v[i].v1);
		int v2=find(v[i].v2);
		if(v1!=v2) {
			for(int j=i+1; j<m; j++) {
				if(v[i].w==v[j].w) {
					int p1=find(v[j].v1);
					int p2=find(v[j].v2);
					if(p1==v1&&p2==v2||p1==v2&&p2==v1)
						f=1;
					else
						break;
				}
			}
			sum=sum+v[i].w;
			merge(v1,v2);
			num++;
		}
	}

	set<int> s;
	for(int i=1; i<=n; i++) {
		s.insert(find(i));
	}
	if(s.size()==1) {
		cout<<sum<<endl;
		if(!f)
			cout<<"Yes"<<endl;
		else
			cout<<"No"<<endl;
	} else {
		cout<<"No MST"<<endl;
		cout<<s.size()<<endl;
	}
	return 0;
}