POJ - 1679 The Unique MST 次小生成树
https://vjudge.net/problem/POJ-1679
很简单的思路,先用kruskal走一遍,并且把用到的边标记,记录此次的最小生成树权值,
然后循环所有的边,对于每一个用过的边不再用它,再来一次 kruskal,看这次的值和第一次的
值是否相等,是则标记并结束循环,不是则直到将所有用过的边循环完为止
AC代码
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=111,M=10010;
int T,n,m;
int p[N];
struct node{
int u,v,w,f;
bool operator< (const node W) const{
return w<W.w;
}
}g[M];
int find(int x)
{
if(p[x]!=x) p[x]=find(p[x]);
return p[x];
}
int kruskal()
{
for(int i=1;i<=n;i++) p[i]=i;
int ans=0,cnt=0;
for(int i=0;i<m;i++)
{
int a=find(g[i].u),b=find(g[i].v);
if(a!=b)
{
p[a]=b;
ans+=g[i].w;
g[i].f=1;
cnt++;
}
if(cnt==n-1) break;
}
if(cnt<n-1) return -1;
else return ans;
}
int kruskal1(int xx)
{
for(int i=1;i<=n;i++) p[i]=i;
int ans=0,cnt=0;
for(int i=0;i<m;i++)
{
if(i==xx) continue;
int a=find(g[i].u),b=find(g[i].v);
if(a!=b)
{
p[a]=b;
ans+=g[i].w;
cnt++;
}
if(cnt==n-1) break;
}
if(cnt<n-1) return -1;
else return ans;
}
int main()
{
cin>>T;
while(T--)
{
cin>>n>>m;
for(int i=0;i<m;i++)
{
scanf("%d %d %d",&g[i].u,&g[i].v,&g[i].w);
g[i].f=0;
}
sort(g,g+m);
int t1=kruskal();
bool flag=true;
int t2;
for(int i=0;i<m;i++)
{
if(g[i].f)
{
t2=kruskal1(i);
if(t2==t1)
{
flag=false;
break;
}
}
}
if(flag) printf("%d\n",t1);
else printf("Not Unique!\n");
}
return 0;
}
本文来自博客园,作者:斯文~,转载请注明原文链接:https://www.cnblogs.com/zhiweb/p/15483307.html

浙公网安备 33010602011771号