次小生成树例题
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;
}