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

题目 https://www.acwing.com/problem/content/239/

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

考虑一个约束满足问题的简化版本:假设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≤10000001≤n≤1000000
1≤i,j≤10000000001≤i,j≤1000000000

输入样例:

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

输出样例:

NO
YES

相等就加入集合,不相等就存起来;

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

AC代码:

#include
#include     //EOF,NULL
#include    //memset
#include    //rand,srand,system,itoa(int),atoi(char[]),atof(),malloc
#include           //ceil,floor,exp,log(e),log10(10),hypot(sqrt(x^2+y^2)),cbrt(sqrt(x^2+y^2+z^2))
#include  //fill,reverse,next_permutation,__gcd,
#include
#include
#include
#include
#include
#include
#include             //setw(set_min_width),setfill(char),setprecision(n),fixed,
#include
#include
#include
#include     //INT_MAX
#include // bitset n
using namespace std;

typedef long long LL;
typedef long long ll;
typedef pair 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+7;
const int MAXN = 1000000+10 ;
int tot,cnt,flag;
int t,n;
int a,b,c;
int pre[MAXN * 2];
int wrong1[MAXN],wrong2[MAXN];
map 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 = 0;
    tot = 0;
    memset(wrong1,0,sizeof wrong1);
    memset(wrong2,0,sizeof wrong2);
    read(n);
    for(int i = 1;i <= n * 2; i++)
      pre[i] = i ;
    H.clear();
    for(int i = 0; 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 = 1;
    for(int i = 0; i < tot; i++){
      if(find(wrong1[i]) == find(wrong2[i])){
        flag = 0;
        break;
      }
    }
    if(flag) printf("YES\n");
    else printf("NO\n");
  }
}

 

你可能感兴趣的:(并查集,算法竞赛进阶指南,数据结构进阶)