判断唯一最小生成树 (克鲁斯卡尔算法)

M . 进阶实验6-3.6：最小生成树的唯一性 [ 问题 9341 ] [ 讨论 ]
Description
给定一个带权无向图，如果是连通图，则至少存在一棵最小生成树，有时最小生成树并不唯一。本题就要求你计算最小生成树的总权重，并且判断其是否唯一。

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

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

Samples
Input 复制
5 7
1 2 6
5 1 1
2 3 4
3 4 3
4 1 7
2 4 2
4 5 5
Output
11
Yes

用Kruskal算法创建最小生成树，如果最小生成树如果不唯一，那就说明最小生成树中的某条边可以换成其他一条同权值的边且保证仍然是最小生成树，如此只需要对最小生成树中权值不唯一的边进行删除并重新进行最小生成树的查找即可。

#include <map>
#include <stack>
#include <queue>
#include <vector>
#include <math.h>
#include <string>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <algorithm>
#include <time.h>
#include <set>
#include <list>
#include <iostream>

using namespace std;
//map<string,string>ip;
//map<string,bool>cy; 

typedef long long ll;
const int maxn=500+188;
const ll inf=1e11;
ll n,m,k,flag,cnt,sum;
int ans;
struct node{
	int x,y,w;
	//bool operator <(node a) const{
	//	return w<a.w;
	//}
}e[260000];
int p[10000];
ll tag[100000],tn;
bool in[600];

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


bool cmp(node a,node b ){
	return a.w<b.w;
}
bool check() {
    for(int i = 0;i < tn;i ++) {
        for(int s=1;s<=n;s++) p[s]=s;
        int d = 0,ee = 0;
        for(int j = 0;j < m;j ++) {
            if(j == tag[i]) continue;
            if(find(e[j].x) != find(e[j].y)) {
                p[find(e[j].x)] = find(p[e[j].y]);
                d += e[j].w;
                ee ++;
            }
        }
        if(ee == n - 1 && d == sum) return true;
    }
    return false;
}
int main() {
    scanf("%d%d",&n,&m);
   	for(int i=1;i<=n;i++) p[i]=i;
    for(int i = 0;i < m;i ++) {
        scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].w);
    }
    sort(e,e + m,cmp);
    for(int i = 0;i < m;i ++) {
        if(find(e[i].x) != find(e[i].y)) {
            p[find(e[i].x)] = find(e[i].y);
            sum += e[i].w;
            if(i < m - 1 && e[i].w == e[i + 1].w || i && e[i].w == e[i - 1].w) {
                flag = 1;
                tag[tn ++] = i;
            }
        }
    }
    for(int i = 1;i <= n;i ++) {
        if(find(i) == i) ans ++;
    }
    if(ans != 1) printf("No MST\n%d",ans);
    else printf("%d\n%s",sum,!flag || !check() ? "Yes" : "No");
}