程式自動分析
題目傳送門
解題思路
先排序
把所有e=1的操作放在前面
然後再進行e=0的操作
在進行e=1的操作的時候
我們隻要把它限制的兩個變量放在同一個集合裡面即可
在e=0,即存在一條不相等的限制條件,
于它限制的兩個變量
如果在一個集合裡面
那就不可能滿足
如不相等的限制條件都滿足
那就YES
資料太大,建議用離散化
離散化
AC代碼
#include<algorithm>
#include<cstdio>
using namespace std;
int tot,f[1000005],b[100005*3];
struct node
{
int x,y,z;
}a[1000005];
bool cmp(node x,node y)
{
return x.z>y.z;
}
int find(int x)
{
if(f[x]==x)return x;
return f[x]=find(f[x]);
}
int main()
{
int n,t;
scanf("%d",&t);
while(t--)
{
int ok=1;
tot=0;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
b[++tot]=a[i].x;
b[++tot]=a[i].y;
}
sort(b,b+tot);
int m=unique(b,b+tot)-b;//去重
for(int i=1;i<=n;i++)
{
a[i].x=lower_bound(b,b+m,a[i].x)-b;
a[i].y=lower_bound(b,b+m,a[i].y)-b;
}
for(int i=1;i<=m;i++)f[i]=i;//初值
sort(a+1,a+n+1,cmp);//排序
for(int i=1;i<=n;i++)
{
int xx=find(a[i].x),yy=find(a[i].y);
if(a[i].z)f[xx]=yy;
else if(xx==yy)
{
printf("NO\n");
ok=0;
break;
}
}
if(ok)printf("YES\n");
}
return 0;
}