POJ - 1679 The Unique MST 次小生成树

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

很简单的思路,先用kruskal走一遍,并且把用到的边标记,记录此次的最小生成树权值,
然后循环所有的边,对于每一个用过的边不再用它,再来一次 kruskal,看这次的值和第一次的
值是否相等,是则标记并结束循环,不是则直到将所有用过的边循环完为止

其他大佬用prim的做法

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;
}
posted @ 2021-07-26 17:52  斯文~  阅读(13)  评论(0)    收藏  举报

你好!