K - The Unique MST POJ - 1679

从前有两只猫。它们实在不知道该出什么题了。


于是它们放弃了治疗，开始玩一个游戏：从乡镇地图中已有的道路里面删除一些道路，并且删除完毕后图仍然是连通的。在所有方案中，删除道路总长度最大的方案为最优方案。


两只猫同时完成了这个游戏。它们都坚信自己是赢家。已知它们的完成方式不同，请判断有没有可能它们的实现方案都是最优的。

 

Input

第一行是一个整数 t (1 <= t <= 20), 测试用例的数量。每个用例代表一张图，第一行是n和m (1 <= n <= 100), 分别为城镇数和道路数。接下来m行为m个三元组 (xi, yi, wi)，表示编号为xi和yi的城镇被长度为wi的道路连接。两个城镇之间最多被一条道路连接。

Output

   对于每个用例，如果答案为否定（即不可能都是最优方案），输出最优方案剩余的（注意不是删除的）道路总长度。否则输出字符串 'Not Unique!'（不带引号）。

Sample Input

2
3 3
1 2 1
2 3 2
3 1 3
4 4
1 2 2
2 3 2
3 4 2
4 1 2

Sample Output

3
Not Unique!

 

题解：这题要用到求次小生成树的方法。具体就是在求最小生成树的过程中把从点i到j的生成树上的边的最大权值记录下来，用maxw[i][j]表示，那么在最小生成树上如果再加任意一条边的话都会形成环，于是要让图成为生成树的话就必须删掉一条边（除加上的边以为），那么删的这条边必须是原来i,j两点的最大边，才有可能次小，如果求出来的要删的次小边与新添的边长度一样，那么就说明最小生成树不止一个。

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int inf = 0x3f3f3f3f;
const int maxn = 110;
int g[maxn][maxn],d[maxn],vis[maxn],link[maxn][maxn],maxw[maxn][maxn],pre[maxn],n,m;
//link[i][j] 判断i,j是否可以连通,pre[i]表示在生成树中i的前一点是哪个

int prim()
{
    memset(d,inf,sizeof(d));
    memset(vis,0,sizeof(vis));
    memset(maxw,0,sizeof(maxw));
    for(int i = 1;i <= n;i++) pre[i] = 1;
    d[1] = 0;

    int sum = 0;
    while(1)
    {
        int v = -1;
        for(int u = 1;u <= n;u++)
        {
            if(!vis[u] && (v == -1 || d[v] > d[u])) v = u;
        }
        if(v == -1) break;
        vis[v] = 1;
        sum += d[v];
        int pred = pre[v];
        maxw[pred][v] = d[v];
        for(int i = 1;i <= n;i++)//更新maxw[i][v]
        {
            maxw[i][v] = max(maxw[pred][v],maxw[i][pred]);
        }
        link[pred][v] = link[v][pred] = 0;//已经在树中的点要标记为不可连通
        for(int u = 1;u <= n;u++)
        {
            if(!vis[u] && d[u] > g[v][u])
            {
                d[u] = g[v][u];
                pre[u] = v;
            }
        }
    }
    return sum;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        memset(link,0,sizeof(link));
        memset(g,inf,sizeof(g));

        scanf("%d %d",&n,&m);
        for(int i = 1;i <= m;i++)
        {
            int u,v,w;
            scanf("%d %d %d",&u,&v,&w);
            g[u][v] = g[v][u] = w;
            link[u][v] = link[v][u] = 1;
        }
        int ans = prim();
        int flag = 0;

        for(int i = 1;i <= n;i++)
        {
            for(int j = 1;j <= n;j++)
            {
                if(!link[i][j] || g[i][j] == inf) continue;
                if(g[i][j] == maxw[i][j])//要删的边和加的边权值相等
                    flag = 1;
            }
        }

        if(flag)
            printf("Not Unique!\n");
        else
            printf("%d\n",ans);
    }
    return 0;
}