求调部分分
查看原帖
求调部分分
824865
lraM41楼主2023/11/25 12:40

写的是非离散化的70pts,实际50pts。

#include<bits/stdc++.h>
using namespace std;
int t,n,fa[1000005],size[1000005];
struct node{
	int x,y;
	bool e;
}a[1000005];
int find(int x){
	return x==fa[x]?x:fa[x]=find(fa[x]);
}
void join(int x,int y)
{
   int f1=find(x),f2=find(y);
   if(f1!=f2){
      if(size[f1]<size[f2])
         swap(f1,f2);
      fa[f2]=f1;
      size[f1]+=size[f2];
   }
}
bool cmp(node s,node t){
	if(s.e==t.e){
		if(s.x==t.x) return s.y<t.y;
		return s.x<t.x;
	}
	return s.e>t.e;
}
void solve(){
	memset(fa,0,sizeof fa);
	memset(a,0,sizeof a);
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i].x>>a[i].y>>a[i].e,fa[i]=i;
	sort(a+1,a+n+1,cmp);
	for(int i=1;i<=n;i++){
		int s=find(a[i].x),t=find(a[i].y);
		if(a[i].e){
			if(s!=t) join(s,t);
		}else if(!a[i].e&&s==t){puts("NO");return;}
	}
	puts("YES");
	return;
}
signed main(){
	cin>>t;
	for(;t;t--) solve();
	return 0;
}
2023/11/25 12:40
加载中...