次小生成树

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
const int maxn = 111;
const int inf = 0x3f3f3f3f;
int Map[maxn][maxn];//邻接矩阵存图
int Max[maxn][maxn];//表示最小生成树中i到j的最大边权
bool used[maxn][maxn];//判断该边是否加入最小生成树，来记录最小生成树的那个边集
int pre[maxn]; // 父亲数组
int dis[maxn]; // 距离数组
void init(int n) // 初始化map数组
{
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= n; j++)
            if (i == j) Map[i][j] = 0;
            else Map[i][j] = inf;
}
void read(int m) // 读入边
{
    int u, v, w;
    for (int i = 0; i < m; i++)
    {
        scanf("%d %d %d", &u, &v, &w);
        Map[u][v] = Map[v][u] = w;
    }
}
int prim(int n)
{
    int ans = 0; 
    bool vis[maxn]; // 标记数组
    memset(vis, false, sizeof(vis)); // 标记数组初始化
    memset(used, false, sizeof(used)); // 最小生成树标记数组
    memset(Max, 0, sizeof(Max)); // 最小瓶颈路数组
    for (int i = 2; i <= n; i++) // 更新dis数组
    {
        dis[i] = Map[1][i];
        pre[i] = 1; //
    }
    pre[1] = 0; // 1本身
    dis[1] = 0;
    vis[1] = true;
    for (int i = 2; i <= n; i++) // 考察剩下的 n - 1个点
    {
        int min_dis = inf, k;
        for (int j = 1; j <= n; j++) // 寻找下一个点
        {
            if (!vis[j] && min_dis > dis[j])
            {
                min_dis = dis[j];
                k = j;
            }
        }
        if (min_dis == inf) return -1;//如果不存在最小生成树
        ans += min_dis;
        vis[k] = true; // 将找到的点加入边集
        used[k][pre[k]] = used[pre[k]][k] = true; // 最小生成树边集
        for (int j = 1; j <= n; j++) // 更新dis数组和最小瓶颈路
        {
            if (vis[j]) Max[j][k] = Max[k][j] = max(Max[j][pre[k]], dis[k]); //更新最小瓶颈路
            if (!vis[j] && dis[j] > Map[k][j])
            {
                dis[j] = Map[k][j];
                pre[j] = k; // 更新父亲节点 
            }
        }
    }
    return ans;//最小生成树的权值之和
}
int smst(int n, int min_ans)//min_ans 是最小生成树的权值和
{
    int ans = inf;
    for (int i = 1; i <= n; i++)//枚举最小生成树之外的边
        for (int j = i + 1; j <= n; j++)
            if (Map[i][j] != inf && !used[i][j])  //有边并且不在最小生成树集合中
                ans = min(ans, min_ans + Map[i][j] - Max[i][j]);// 加边成环，减去最小瓶颈路
    if (ans == inf) return -1;
    return ans;
}
void solve(int n)
{
    int ans = prim(n);
    if (ans == -1)
    {
        puts("Not Unique!");
        return;
    }
    if (smst(n, ans) == ans)
    printf("Not Unique!\n");
    else
    printf("%d\n", ans);
}

int main()
{
    int T, n, m;
    scanf("%d", &T);
    while (T--)
    {
        scanf("%d %d", &n, &m);
        init(n);
        read(m);
        solve(n);
    }
    return 0;
}