次小生成树kruskal算法个人理解POJ 1679 The Unique MST

次小生成树例题

POJ 1679 The Unique MST

算法核心

在原有最小生成树的基础上，枚举每一条不在这个树上的边，将它加入到这个最小生成树当中去，那么就会形成一个环（最小生成树当中一定无环），取这个环中最大的边（除了新加的之外），剔除之后求总权值，不断取min，取到最后即是次短

代码（也可作板子）：

/**
 *  Author1: low-equipped w_udixixi
 *  Author2: Sher丶lock
 *  Date :2019-08-15
 **/
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<queue>
#include<map>
#define ll long long
#define pb push_back
#define rep(x,a,b) for (int x=a;x<=b;x++)
#define repp(x,a,b) for (int x=a;x<b;x++)
#define pi 3.14159265358979323846
#define mem(a,x) memset(a,x,sizeof a)
using namespace std;
const int maxn=2e4+7;
const int INF=1e9;
const ll INFF=1e18;
int n,m;
struct Edge{
    int u,v,w;
    bool vis;
}p[maxn];
vector<int > G[maxn];
int fa[maxn];
int MAX[300][300];//表示i到j在最小生成树上的最大路径
bool cmp(Edge a,Edge b)
{
    return a.w<b.w;
}
int find(int x)
{
    if (fa[x]!=x)
    {
        return fa[x]=find(fa[x]);
    }
    else return x;
}
void kruskal()
{
    sort(p,p+m,cmp);
    rep(i,0,n)
    {
        G[i].clear();
        G[i].pb(i);
        fa[i]=i;
    }
    int ans=0;
    int cnt=0;
    repp(i,0,m)
    {
        int t1=find(p[i].u);
        int t2=find(p[i].v);
        if (t1!=t2)
        {
            cnt++;
            p[i].vis=true;
            ans+=p[i].w;
            int x1=G[t1].size();
            int x2=G[t2].size();
            repp(j,0,x1)
            {
                repp(k,0,x2)//由于p[i].w递增，所以这个更新最大值可行
                {
                    MAX[G[t1][j]][G[t2][k]]=MAX[G[t2][k]][G[t1][j]]=p[i].w;
                }
            }
            fa[t1]=t2;
            repp(j,0,x1)//t2变成t1的父亲，t1的所有儿子也归t2
            {
                G[t2].pb(G[t1][j]);
            }
        }
        if (cnt==n-1)break;
    }
    int cians=INF;//次小权值
    repp(i,0,m)
    {
        if (!p[i].vis)
        {
            cians=min(cians,ans+p[i].w-MAX[p[i].u][p[i].v]);
            //从最小生成树中取出一条非访问边（即不在最小生成树上的边）
            //加入其中，并删掉环中最大的边
        }
    }
    if (cians>ans)cout<<ans<<endl;
    else cout<<"Not Unique!"<<endl;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        repp(i,0,m)
        {
            scanf("%d%d%d",&p[i].u,&p[i].v,&p[i].w);
            p[i].vis=false;
        }
        kruskal();
    }
    return 0;
}