求调最小生成树
查看原帖
求调最小生成树
936249
zhangqiuyanAFOon2024楼主2024/8/7 10:43

题目

代码:

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define int long long
const int N=1e4+5;
int fa[N],n,m;
void init(){
	for(int i=1;i<=n;i++) fa[i]=i;
}
struct node{
	int u,v,w;
//	bool operator <(const node &a) const{return w<a.w;}
}e[N];
int findp(int x){
	return x==fa[x]?x:fa[x]=findp(fa[x]);
}
void merge(int u,int v){
	fa[findp(u)]=findp(v);
}
bool cmp(node x,node y){
	return x.w<y.w;
}
bool vis[N];
void solve(){
	cin>>n>>m;
	for(int i=1;i<=m;i++)
		cin>>e[i].u>>e[i].v>>e[i].w,vis[i]=0;
	sort(e+1,e+m+1,cmp);
	int cnt=0,ans=0;
	init();
	for(int i=1;i<=m;i++){
		int nowu=e[i].u,nowv=e[i].v;
		if(findp(nowu)!=findp(nowv)){
			merge(findp(nowu),findp(nowv));
			cnt++;
			ans+=e[i].w;
			vis[i]=1;
		}
		if(cnt==n-1) break;
	}
	for(int i=1;i<=m;i++){
		if(vis[i]){
			init();	
			cnt=0;
			int t=0;
			for(int j=1;j<=m;j++){
				if(i==j) continue;
				int nowu=e[j].u,nowv=e[j].v;
				if(findp(nowu)!=findp(nowv)){
					merge(findp(nowu),findp(nowv));
					cnt++;
					t+=e[j].w;
				}
				if(cnt==n-1){
					if(t==ans){
						puts("Not Unique!");
						return;
					}
					break;
				}
			}
		}
	}
	cout<<ans<<endl;
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	int T;
	cin>>T;
	while(T--) solve(); 
	return 0;
}
2024/8/7 10:43
加载中...