poj1679 The Unique MST(次小生成树)

http://poj.org/problem?id=1679

给出一些边，求最小生成树是不是唯一的。

思路就是求次小生成树与最小生成树是否相同。

如何求次小生成树呢，假设已经有n-1条边被选择了，我们就需要用别的没有用上的边来替换已选择的边。

用一个二维数组记录点与点之间权值最大的边，用这2点之间没用上的边来换掉，这样就可以求得一个新的生成树的长度。

我们求这个值的最小，即次小生成树的长度。

然后比较一下次小生成树与最短生成树是否相同即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
bool vis[111];
bool used[111][111];
int pre[111];
int ma[111][111];
int dis[111];
int big[111][111];
int n,m;
int prim()
{
	pre[1]=0;
	for(int i=2;i<=n;i++)
	pre[i]=1;
	dis[1]=0;
	for(int i=2;i<=n;i++)
	dis[i]=ma[1][i];
	vis[1]=1;
	int ans=0;
	for(int i=1;i<=n-1;i++)
	{
		int mi=9999999;
		int p;
		for(int j=1;j<=n;j++)
		{
			if(!vis[j]&&dis[j]<mi)
			mi=dis[j],p=j;
		}
		if(mi==9999999)return -1;
		ans+=mi;
		vis[p]=1;
		used[p][pre[p]]=used[pre[p]][p]=1;
		for(int j=1;j<=n;j++)
		{
			if(vis[j])
			big[j][p]=big[p][j]=max(big[j][pre[p]],mi);
			if(!vis[j]&&ma[p][j]<dis[j])
			{
				dis[j]=ma[p][j];
				pre[j]=p;
			}
		}
	}
	return ans;
}
int cmst(int ans)
{
     int mi=99999999;
     for(int i=1;i<=n;i++)
     for(int j=i+1;j<=n;j++)
     {
     	if(ma[i][j]!=0x3f3f3f&&!used[i][j])
     	{
     		mi=min(mi,ans+ma[i][j]-big[i][j]);
		 }
	 }
	 if(mi==99999999)return -1;
	 return mi;
}
int main(){
	int t;
	cin>>t;
	while(t--)
	{
		memset(used,0,sizeof(used));
		memset(big,0,sizeof(big));
		memset(vis,0,sizeof(vis));
		memset(dis,0x3f3f3f,sizeof(dis));
		memset(ma,0x3f3f3f,sizeof(ma));
		cin>>n>>m;
		for(int i=1;i<=n;i++)
		 ma[i][i]=0;
		int u,v,w;
		for(int i=0;i<m;i++)
		{
			cin>>u>>v>>w;
			ma[u][v]=ma[v][u]=min(ma[u][v],w);
		}
		int ans=prim();
		if(ans==-1||ans==cmst(ans))
		{
			cout<<"Not Unique!"<<endl;
		}
		else cout<<ans<<endl;
	}
	return 0;
}