程序自动分析（prog）

程序自动分析（prog）

时间限制: 2 Sec 内存限制: 512 MB
题目描述
在实现程序自动分析的过程中，常常需要判定一些约束条件是否能被同时满足。
考虑一个约束满足问题的简化版本：假设x1,x2,x3,…代表程序中出现的变量，给定n个形 xi=xj 或 xi≠xj 的变量相等/不等的约束条件，请判定是否可以分别为每一个变量赋予恰当的值，使得上述所有约束条件同时被满足。例如，一个问题中的约束条件为：x1=x2，x2=x3，x3=x4，x1≠x4，这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。
现在给出一些约束满足问题，请分别对它们进行判定。
输入
输入文件的第 1 行包含 1 个正整数 t，表示需要判定的问题个数。注意这些问题之间是相互独立的。
对于每个问题，包含若干行：
第 1 行包含 1 个正整数 n，表示该问题中需要被满足的约束条件个数。
接下来 n 行，每行包括 3 个整数 i,j,e，描述 1 个相等/不等的约束条件，相邻整数之间用单个空格隔开。若 e=1，则该约束条件为 xi=xj；若 e=0，则该约束条件为 xi≠xj。
输出
输出文件包括 t 行。
输出文件的第 k 行输出一个字符串 “YES” 或者 “NO”（不包含引号，字母全部大写），“YES” 表示输入中的第 k 个问题判定为可以被满足，“NO” 表示不可被满足。
样例输入一
2
2
1 2 1
1 2 0
2
1 2 1
2 1 1
样例输入二
2
3
1 2 1
2 3 1
3 1 1
4
1 2 1
2 3 1
3 4 1
1 4 0
样例输出一
NO
YES
样例输出二
YES
NO
数据范围
1<=n<=100000
来源
NOI2015 Day1

题解

把相等关系转化为并查集随便搞搞就好了。。（ps.NOI出这种题我也是醉了）

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#define N 100010
using namespace std;
int T,n,tot,c[N*2],fa[N*2];
struct node
{
  int a,b,c;
  bool operator<(const node x)
  const{return c>x.c;}
}t[N];

int find(int x)
{
  if(fa[x]==x)return x;
  return fa[x]=find(fa[x]);
}

int main()
{
  int a,b,flag;
  scanf("%d",&T);
  while(T--)
  {
    scanf("%d",&n);
    flag=0;tot=0;
    for(int i=1;i<=n;i++)
    {
      scanf("%d%d%d",&t[i].a,&t[i].b,&t[i].c);
      c[++tot]=t[i].a;c[++tot]=t[i].b;
    }
    sort(t+1,t+n+1);
    sort(c+1,c+tot+1);
    int Tnum=unique(c+1,c+tot+1)-c-1;
    for(int i=1;i<=Tnum;i++)fa[i]=i;
    for(int i=1;i<=n;i++)
    {
      a=lower_bound(c+1,c+Tnum+1,t[i].a)-c;
      b=lower_bound(c+1,c+Tnum+1,t[i].b)-c;
      if(t[i].c)fa[find(b)]=find(a);
      else if(find(a)==find(b)){flag=1;break;}
    }
    if(flag)printf("NO\n");
    else printf("YES\n");
  }
  return 0;
}