【NOI2015】BZOJ4195程序自动分析（并查集+离散化）

4195: [Noi2015]程序自动分析

Time Limit: 10 Sec Memory Limit: 512 MB

Submit: 836 Solved: 361

[Submit][Status][Discuss]

Description

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

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

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

Input

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

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

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

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

Output

输出文件包括t行。

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

Sample Input

2

2

1 2 1

1 2 0

2

1 2 1

2 1 1

Sample Output

NO

YES

HINT

在第一个问题中，约束条件为：x1=x2,x1≠x2。这两个约束条件互相矛盾，因此不可被同时满足。

在第二个问题中，约束条件为：x1=x2,x2=x1。这两个约束条件是等价的，可以被同时满足。

1≤n≤1000000

1≤i,j≤1000000000

NOI的题越来越水了，先处理e=1时的情况，用离散化处理大数据，最后并查集秒出结果。我的离散化特别SB，想看离散化的去这博客。。http://blog.csdn.net/dad3zz

P.S.我的程序只有在BZOJ上能过，其他OJ只能过九组，最后一组T，因为BZOJ测的是总时间。。

NOIP初赛小朋友都会做。。

#include<cstdio>
#include<algorithm>
using namespace std;
int t,n,i,j,m,father[2000001];
struct edge
{
int x,y,e;
};
edge a[1000001];
bool flag;

int find(int s)
{
int p,q;
p=s;
while (p!=father[p])
p=father[p];
while (s!=father[s])
{
q=s;
s=father[s];
father[q]=p;
}
return s;
}

void merge(int x1,int y1)
{
int f1=find(x1);
int f2=find(y1);
if (f1!=f2)
father[f1]=f2;
}

bool cmp(const edge x1,const edge x2)
{
return x1.e>x2.e;
}

int main()
{
freopen("prog.in","r",stdin);
freopen("prog.out","w",stdout);
scanf("%d",&t);
for (i=1;i<=t;i++)
{
flag=true;
scanf("%d",&n);
for (j=1;j<=2000000;j++)
father[j]=j;
for (j=1;j<=n;j++)
{
scanf("%d %d %d",&a[j].x,&a[j].y,&a[j].e);
if (a[j].x>1000000)
a[j].x=1000000+a[j].x/1000000+a[j].x%1000000;
if (a[j].y>1000000)
a[j].y=1000000+a[j].y/1000000+a[j].y%1000000;
}
sort(a+1,a+n+1,cmp);
for (j=1;j<=n;j++)
{
if (a[j].e==1)
merge(a[j].x,a[j].y);
if (a[j].e==0)
if (find(a[j].x)==find(a[j].y))
{
flag=false;
break;
}
}
if (flag)
printf("%s\n","YES");
else
printf("%s\n","NO");
}
fclose(stdin);
fclose(stdout);
return 0;
}