五彩斑斓求调
查看原帖
五彩斑斓求调
739307
All_Failed_OI_Cici楼主2024/8/28 20:01

qwq

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int fa[N];
int n,t;
struct node{
	int e,a,b;
}a[N];
int get(int x){
	return x==fa[x]?x:get(fa[x]);
}
void merge(int x,int y){
	fa[get(y)]=get(x);
}
bool fi(int a,int b){
	return fa[a]==fa[b]?true:false;
}
void init(){
	memset(a,0,sizeof(a));
	memset(fa,0,sizeof(fa));
	for(int i=1;i<=n;i++) fa[i]=i;	
}
bool cmp(node x,node y){
	return x.e>y.e;
}
signed main(){
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		init();
		for(int i=1;i<=n;i++) scanf("%d%d%d",&a[i].a,&a[i].b,&a[i].e);
		sort(a+1,a+n+1,cmp);
		int i;
		bool flag=true;
		for(i=1;a[i].e==1;i++) merge(a[i].a,a[i].b);
		for(;i<=n;i++){
			if(fi(a[i].a,a[i].b)){
				puts("NO"),flag=false;
				break;
			}
		}
		if(flag) puts("YES");
	}
    return 0;
}

注:样例均通过

2024/8/28 20:01
加载中...