POJ1679：次小生成树

POJ1679

题解：https://blog.csdn.net/weixin_42165981/article/details/81946849。

• 设最小生成树为T，非T当中的边集为E。
• 我们每次要在E中选一条边a，在T中加入边a(v1,v2)之后肯定形成一个环，那么我们要去掉在环中原来属于T的最长边b。
• 如何求b？我们在使用kruskal算法时，遇到v1和v2属于两个不同的并查集，那么添加的边一定是最长的边把两个集合联系起来。所以我们保存这一条边即可。

代码：

#include <cstring>
#include <algorithm>
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
int const M = 20000+10;
int const N = 100 + 10;
int const inf = 0x3f3f3f3f;
int n,m,fa[N],maxd[N][N],vis[M];
struct Edge
{
	int u,v,w;
}p[M];
vector<int>G[N];
bool cmp(Edge a,Edge b){
	return a.w < b.w;
}
int find(int x){
	return x == fa[x] ? x : (fa[x] = find(fa[x]));
}
void Kruskal(){
	sort(p,p+m,cmp);
	for(int i=0;i<=n;i++){
		G[i].clear();
		G[i].push_back(i);
		fa[i] = i;
	}
	int sum = 0,e = 0;   //记录最小生成树的权值,e记录边数
	for(int i=0;i<m;i++){
		if(e == n-1)	break;  //如果边数有n-1就停止
		int x = find(p[i].u),	y = find(p[i].v);
		if(x != y){
			e++;   //每次添加边
			vis[i] = true;  //访问过边了
			sum += p[i].w;   //统计权值和
			for(int j=0;j<G[x].size();j++)  
				for(int k=0;k<G[y].size();k++)
					maxd[G[x][j]][G[y][k]] = maxd[G[y][k]][G[x][j]] = p[i].w;
			fa[x] = y;
			for(int j=0;j<G[x].size();j++)	G[y].push_back(G[x][j]);
		}
	}
	int MIN = inf;
	for(int i=0;i<m;i++)
		if(!vis[i])
			MIN = min(MIN,sum+p[i].w-maxd[p[i].u][p[i].v]);
	if(MIN > sum)	printf("%d\n",sum);
	else printf("Not Unique!\n");
}
int main(){
	int T;
	scanf("%d\n",&T);
	while(T--){
		scanf("%d%d",&n,&m);
		for(int i=0;i<m;i++){
			scanf("%d%d%d",&p[i].u,&p[i].v,&p[i].w);
			vis[i] = false;
		}
		Kruskal();
	}
	return 0;
}