7-20 最小生成树的唯一性 (35 point(s))

7-20 最小生成树的唯一性 (35 point(s))

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

输入格式：

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

输出格式：

如果存在最小生成树，首先在第一行输出其总权重，第二行输出“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

代码：

/*
将每个边的信息存储在向量中，依旧从小到大排列，依次遍历判断是否加入此边（记为a）不是环路（起点终点的f值不同），并判断下一条边（记为b）是否权值与此边相同并且a边的两端各自与b的两端在一棵树上（起点代表都在已成的树上 终点为都可连接到接下来的最小树上），若满足，则代表树不唯一。若边数为n-1则已成，否则没有最小支撑树。
*/
#include <bits/stdc++.h>
using namespace std;
const int N = 510;

struct Edge
{
    int a, b, w;
    bool operator<(const Edge &W) const
    {
        return w < W.w;
    }
} edge[N*N];
int f[N];
int cnt = 0, res = 0;
int n, m;
bool flag = 0;

int find(int x)
{
    if (x != f[x])
        f[x] = find(f[x]);
    return f[x];
}

bool kruskal()
{
    sort(edge, edge + m);
    for (int i = 1; i <= n; i++)
        f[i] = i;

    for (int i = 0; i < m; i++)
    {
        int a = edge[i].a, b = edge[i].b, w = edge[i].w;
        int t1 = find(a);
        int t2 = find(b);
        if (t1 != t2)
        {
            int j = i + 1;
            if (edge[j].w == edge[i].w)
            {
                int u = find(edge[j].a);
                int v = find(edge[j].b);
                if (u == t1 && v == t2 || u == t2 && v == t1)
                    flag = 1;
            }
            f[t1] = t2;
            cnt++;
            res += w;
        }
    }

    if (cnt == n - 1)
        return true;
    else
        return false;
}

int main()
{
    cin >> n >> m;
    for (int i = 0; i < m; i++)
    {
        int a, b, w;
        cin >> a >> b >> w;
        edge[i] = {a, b, w};
    }

    if (kruskal())
    {
        cout << res << endl;
        if (flag)
            cout << "No";
        else
            cout << "Yes";
    }
    else
    {
        cout << "No MST" << endl;
        int ans = 0;
        for (int i = 1; i <= n; i++)
            if (f[i] == i)
                ans++;
        cout << ans;
    }

    return 0;
}