237. 程序自动分析 【map+并查集】

程序自动分析

描述

在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。

考虑一个约束满足问题的简化版本：假设x1,x2,x3,…x1,x2,x3,…代表程序中出现的变量，给定n个形如xi=xjxi=xj或xi≠xjxi≠xj的变量相等/不等的约束条件，请判定是否可以分别为每一个变量赋予恰当的值，使得上述所有约束条件同时被满足。

例如，一个问题中的约束条件为：x1=x2，x2=x3，x3=x4，x1≠x4x1=x2，x2=x3，x3=x4，x1≠x4，这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。

现在给出一些约束满足问题，请分别对它们进行判定。

输入格式

输入文件的第1行包含1个正整数t，表示需要判定的问题个数，注意这些问题之间是相互独立的。

对于每个问题，包含若干行：

第1行包含1个正整数n，表示该问题中需要被满足的约束条件个数。

接下来n行，每行包括3个整数i,j,e，描述1个相等/不等的约束条件，相邻整数之间用单个空格隔开。若e=1，则该约束条件为xi=xjxi=xj；若e=0，则该约束条件为xi≠xjxi≠xj。

输出格式

输出文件包括t行。

输出文件的第k行输出一个字符串“YES”或者“NO”（不包含引号，字母全部大写），“YES”表示输入中的第k个问题判定为可以被满足，“NO”表示不可被满足。

数据范围

1≤n≤1000000
1≤i,j≤1000000000

输入样例：

2
2
1 2 1
1 2 0
2
1 2 1
2 1 1

输出样例：

NO
YES

题解：

相等就加入集合，不相等就存起来；

最后判断一下不相等的是否在一个集合；

代码：

#include<iostream>
#include<cstdio>     //EOF,NULL
#include<cstring>    //memset
#include<cstdlib>    //rand,srand,system,itoa(int),atoi(char[]),atof(),malloc
#include<cmath>           //ceil,floor,exp,log(e),log10(10),hypot(sqrt(x^2+y^2)),cbrt(sqrt(x^2+y^2+z^2))
#include<algorithm>  //fill,reverse,next_permutation,__gcd,
#include<string>
#include<vector>
#include<queue>
#include<stack>
#include<utility>
#include<iterator>
#include<iomanip>             //setw(set_min_width),setfill(char),setprecision(n),fixed,
#include<functional>
#include<map>
#include<set>
#include<limits.h>     //INT_MAX
#include<bitset> // bitset<?> n
using namespace std;
 
typedef long long LL;
typedef long long ll;
typedef pair<int,int> P;
#define all(x) x.begin(),x.end()
#define readc(x) scanf("%c",&x)
#define read(x) scanf("%d",&x)
#define read2(x,y) scanf("%d%d",&x,&y)
#define read3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define print(x) printf("%d\n",x)
#define mst(a,b) memset(a,b,sizeof(a))
#define lowbit(x) x&-x
#define lson(x) x<<1
#define rson(x) x<<1|1
#define pb push_back
#define mp make_pair
const int INF =0x3f3f3f3f;
const int mod = 1e9+;
const int MAXN = + ;
int tot,cnt,flag;
int t,n;
int a,b,c;
int pre[MAXN * ];
int wrong1[MAXN],wrong2[MAXN];
map<int,int> H;
int mapping(int x)
{
    if (H.count(x)) return H[x];
    return H[x] = ++ cnt  ;
}
int find(int x){
  return x == pre[x] ? x : pre[x] = find(pre[x]);
}
void join(int x,int y){
  if(find(x) != find(y)){
    pre[find(y)] = find(x);
  }
}
int main(){
  read(t);
  while(t--){
    cnt = ;
    tot = ;
    memset(wrong1,,sizeof wrong1);
    memset(wrong2,,sizeof wrong2);
    read(n);
    for(int i = ;i <= n * ; i++)
      pre[i] = i ;
    H.clear();
    for(int i = ; i < n; i++){
      read3(a,b,c);
      a = mapping(a) ;
      b = mapping(b);
      if(c) {
        join(a,b);
      }
      else{
        wrong1[tot] = a, wrong2[tot] = b;
        tot++;
      }
    }
    flag = ;
    for(int i = ; i < tot; i++){
      if(find(wrong1[i]) == find(wrong2[i])){
        flag = ;
        break;
      }
    }
    if(flag) printf("YES\n");
    else printf("NO\n");
  }
}