The Unique MST POJ - 1679（非严格次小生成树）

题目链接

大意：问由村庄构成的最小生成树是否唯一(这个代码只能判断最小生成树是否唯一，但不能求出严格次小生成树）

严格次小生成树以后要补上

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#define maxn 150

using namespace std;

struct node{
	int u,v,w;
	bool vis;//比最小生成树多了个vis标志 
}edge[maxn*maxn];

int n,m,ans,tot;
int s[maxn],maps[maxn][maxn]; //maps数组用来维护u到v的最大距离，在枚举边的时候会用到
vector<int> p[maxn];//扩展路径 

bool cmp(const node &a,const node &b){
	return a.w<b.w;
}

int find(int u){
	return s[u]==u?u:find(s[u]);
}

void Kruskal(){
	tot=0,ans=0;
	sort(edge,edge+m,cmp);
	for(int i=0;i<m;i++){
		int a=find(edge[i].u);
		int b=find(edge[i].v);
		if(a!=b){
			s[b]=a;
			edge[i].vis=true;//标记入树的集合 
			ans+=edge[i].w;
			tot++;
			for(int j=0;j<p[a].size();j++){
				for(int k=0;k<p[b].size();k++){
					maps[p[a][j]][p[b][k]]=maps[p[b][k]][p[a][j]]=edge[i].w;
				}
			}
			for(int j=0;j<p[b].size();j++){
				p[a].push_back(p[b][j]);//因为路径压缩了，所以要把点存进父节点 
			}
		}
		if(tot==n){
			break;
		}
	}
}

int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		memset(maps,0,sizeof maps);
		scanf("%d%d",&n,&m);
		for(int i=0;i<=n;i++){
			s[i]=i;
			p[i].clear();
			p[i].push_back(i);
		}
		for(int i=0;i<m;i++){
			scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
			edge[i].vis=false; 
		} 
		Kruskal();
		int dis=0x3f3f3f3f;
		for(int i=0;i<m;i++){
			if(!edge[i].vis){
				dis=min(dis,ans+edge[i].w-maps[edge[i].u][edge[i].v]);
			}
		}
		if(dis>ans){//判断次小生成树和最小生成树 
			printf("%d\n",ans);
		}else{
			printf("Not Unique!\n");
		}
	}
	return 0;
}