POJ ~ 1679 ~ The Unique MST （次小生成树）

题意：T组测试数据，有N个点M条边，问最小生成树是否唯一？唯一输出权重，不唯一输出“Not Unique!”。

思路：求次小生成树。其实求法很简单，就是记录一下最小生成树用的边，然后尝试删去这些边中的某一条边，然后再求最小生成树。

数据有点坑，代码下面附3组数据。

 

//#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int MAXN = 1005;
const int MAXM = 1005 * 1005;
int n, s, m, f[MAXN];
bool flag;
struct Point
{
    int x, y;
}point[MAXN];
struct Edge
{
    int u, v, w;
    bool use, del;//最小生成树用的边，该边是否被删除
    bool operator < (const Edge& A) const
    {
        return w < A.w;
    }
}edge[MAXM];
void init()
{
    for (int i = 0; i <= n; i++) f[i] = i;
}
int Find(int x){ return f[x] == x ? x : f[x] = Find(f[x]); }
int kruskal()
{
    init();
    int ans = 0, cnt = 0;
    for (int i = 0; i < m; i++)
    {
        int u = edge[i].u, v = edge[i].v, w = edge[i].w;
        int root1 = Find(u), root2 = Find(v);
        if (root1 != root2 && edge[i].del == false)
        {
            if (!flag) edge[i].use = true;//最小生成树用的边
            f[root1] = root2;
            ans += w;
            cnt++;
            if (cnt == n - 1) return ans;
        }
    }
    return -1;
}
int main()
{
    int T;
    scanf("%d", &T);
    while (T--)
    {
        m = 0;
        scanf("%d%d", &n, &m);
        if (m == 0) { printf("0\n"); continue; }
        for (int i = 0; i < m; i++)
        {
            scanf("%d%d%d", &edge[i].u, &edge[i].v, &edge[i].w);
            edge[i].use = false;
            edge[i].del = false;
        }
        sort(edge, edge + m);
        flag = false;
        int ans = kruskal();//最小生成树
        flag = true;
        for (int i = 0; i < m; i++)//次小生成树
        {
            if (edge[i].use == true)//最小生成树的边
            {
                edge[i].del = true;//标记为删除
                int t = kruskal();
                if (t == ans) { ans = -1; break; }//最小生成树和次小生成树权值相同
                edge[i].del = false;//标记改回没被删除
            }
        }
        if (ans != -1) printf("%d\n", ans);
        else printf("Not Unique!\n");
    }
    return 0;
}
/*
3
10 15
1 2 14
4 1 89
7 6 8
9 4 81
5 2 81
10 9 6
1 5 44
1 3 33
2 6 25
6 10 10
1 10 65
6 9 74
8 10 41
2 3 89
5 10 2
6 7
1 3 1
1 2 2
2 3 3
3 4 0
4 6 5
4 5 4
5 6 6
1 0
*/
//输出220 12 0