POJ 1679 The Unique MST （两种思路的次小生成树）

https://vjudge.net/problem/POJ-1679

次小生成树模板题

思路1：由于数据比较小，我们把最小生成树每条边记录下来，之后进行k（k = 最小生成树边数）次最小生成树算法，每次都去掉一条边，判断得到的最小是是否等于最小生成树，这样必然会很慢

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<cmath>
#define ll long long
#define mod 1000000007
#define inf 0x3f3f3f3f
using namespace std;
struct node
{
    int u, v, w;
}a[10005];
int pre[105];
int miny[10005];
int n, m;
void init()
{
    for(int i = 1; i <= n; i ++)
        pre[i] = i;
}
int find(int x)
{
    if(x == pre[x]) return x;
    else return find(pre[x]);
}
int join(int a, int b)
{
    int i = find(a);
    int j = find(b);
    if(i == j) return 0;
    else
    {
        pre[i] = j;
        return 1;
    }
}
bool cmp(node a, node b)
{
    return a.w < b.w;
}
int main()
{
    int t;
    scanf("%d", &t);
    while(t --)
    {
        scanf("%d%d", &n, &m);
        init();
        for(int i = 1; i <= m; i ++)
            scanf("%d%d%d", &a[i].u, &a[i].v, &a[i].w);
        sort(a + 1, a + 1 + m, cmp);
        int ans = 0;
        int tot = 0;
        int cnt = 0;
        for(int i = 1; i <= m; i ++)
        {
            if(join(a[i].u, a[i].v))
            {
                ans += a[i].w;
                miny[++ cnt] = i;
                tot ++;
            }
            if(tot == n - 1) break;
        }
        bool uni = 0;
        int ans1 = ans;
        for(int k = 1; k <= cnt; k ++)
        {
            init();
            tot = 0, ans = 0;
            bool flag = 0;
            for(int i = 1; i <= m; i ++)
            {
                if(miny[k] == i) continue;
                if(join(a[i].u, a[i].v))
                {
                    ans += a[i].w;;
                    tot ++;
                }
                if(tot == n - 1)
                {
                    flag = 1;
                    break;
                }
            }
            if(flag && ans == ans1)
            {
                uni = 1;
                break;
            }
        }
        if(! uni)
        {
            printf("%d\n", ans1);
        }
        else printf("Not Unique!\n");
    }
    return 0;
}

思路二：Kruskal次小生成树算法，和最小生成树同理，只不过我们每次都记录下来每两个点之间的最大距离，这样删去这条边再加上一条边，必然构成生成树（证明略）

对于一棵已经求出的最小生成树
枚举每一条不在最小生成树上的边
并把这条边加入最小生成树
设这条边连接的两个节点为u和v
这时树上u到v的路径会出现回路
所以我们需要删掉u到v路径上的一条边使其重新变成一颗树
这时生成树权值的增量就是新加入的边权-删掉的边权
为了使增量最小，我们要删去的自然是u到v路径中权值最大的

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<cmath>
#define ll long long
#define mod 1000000007
#define inf 0x3f3f3f3f
using namespace std;
struct node
{
    int u, v, w;
}a[10005];
int pre[105];
int miny[10005];
int maxy[105][105];
vector<int> G[105];
bool vis[10005];
int n, m;
int ans;
void init()
{
    for(int i = 1; i <= n; i ++)
    {
        G[i].clear();
        G[i].push_back(i);
        pre[i] = i;
    }
    memset(vis, 0, sizeof(vis));
    memset(maxy, -1, sizeof(maxy));
    ans = 0;
}
int find(int x)
{
    if(x == pre[x]) return x;
    else return find(pre[x]);
}
bool cmp(node a, node b)
{
    return a.w < b.w;
}
bool Kruskal()
{
    init();
    int tot = 0;
    sort(a + 1, a + 1 + m, cmp);
    for(int i = 1; i <= m; i ++)
    {
        int u = find(a[i].u);
        int v = find(a[i].v);
        if(u != v)
        {
            tot ++;
            ans += a[i].w;
            vis[i] = 1;
            pre[u] = v;
            int sz1 = G[u].size(), sz2 = G[v].size();
            for(int j = 0; j < sz1; j ++)
                for(int k = 0; k < sz2; k ++)
                    maxy[G[u][j]][G[v][k]] = maxy[G[v][k]][G[u][j]] = a[i].w;
            for(int j = 0; j < sz1; j ++)
                G[v].push_back(G[u][j]);
        }
        if(tot == n - 1) break;
    }
    int sec = inf;
    for(int i = 1; i <= m; i ++)
    {
        if(! vis[i])
        {
            sec = min(sec, ans - maxy[a[i].u][a[i].v] + a[i].w);
        }
    }
    if(ans == sec) return 0;
    return 1;
}
int main()
{
    int t;
    scanf("%d", &t);
    while(t --)
    {
        scanf("%d%d", &n, &m);
        init();
        for(int i = 1; i <= m; i ++)
            scanf("%d%d%d", &a[i].u, &a[i].v, &a[i].w);
        if(Kruskal()) printf("%d\n", ans);
        else printf("Not Unique!\n");
    }
    return 0;
}