题目描述
在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。
考虑一个约束满足问题的简化版本:假设x1,x2,x3…代表程序中出现的变量,给定n个形如xi=xj或xi≠xj的变量相等/不等的约束条件,请判定是否可以分别为每一个变量赋予恰当的值,使得上述所有约束条件同时被满足。
1<=n<=105 1<=x<=109
例如,一个问题中的约束条件为:x1=x2,x2=x3,x3=x4,x4≠x1,这些约束条件显然是不可能同时被满足的,因此这个问题应判定为不可被满足。 现在给出一些约束满足问题,请分别对它们进行判定。
输入格式: 从文件prog.in中读入数据。
输入文件的第1行包含1个正整数t,表示需要判定的问题个数。注意这些问题之间是相互独立的。
对于每个问题,包含若干行: 第1行包含1个正整数n,表示该问题中需要被满足的约束条件个数。
接下来n行,每行包括3个整数i,j,e,描述1个相等/不等的约束条件,相邻整数之间用单个空格隔开。
若e=1,则该约束条件为xi=xj;若e=0,则该约束条件为xi≠xj;
输出格式: 输出到文件 prog.out 中。
输出文件包括t行。 输出文件的第 k行输出一个字符串“ YES” 或者“ NO”(不包含引号,字母全部大写),“ YES” 表示输入中的第k个问题判定为可以被满足,“ NO” 表示不可被满足。
输入输出样例
输入样例#1:
2
2
1 2 1
1 2 0
2
1 2 1
2 1 1
输出样例#1:
NO
YES
思路:离散化,并查集。
【样例说明 2】
在第一个问题中,约束条件有三个:𝑥1 = 𝑥2, 𝑥2 = 𝑥3, 𝑥3 = 𝑥1 。只需赋值使得 𝑥1 = 𝑥2 = 𝑥3 ,即可同时满足所有的约束条件。
在第二个问题中,约束条件有四个: 𝑥1 = 𝑥2, 𝑥2 = 𝑥3, 𝑥3 = 𝑥4, 𝑥1 ≠ 𝑥4 。由前三个约束条件可以推出 𝑥1 = 𝑥2 = 𝑥3 = 𝑥4 ,然而最后一个约束条件却要求𝑥1 ≠ 𝑥4
对于初学者,都不知道为什么要离散化,尤其是并查集题,如果x的范围很大,109。
假如样例输入
1
3
12345678 12345677 1
12345677 12345676 1
12345678 12345676 1
样例的意思就是
x12345678=x12345677
x12345677=x12345676
x12345678=x12345676
上面只用到3个变量,但是在初始化时得开一个很大的数组
int fa[12345679];
for(int i=1;i<=12345678;i++)
fa[i]=i;
开这么大数组,只用到三个变量,离散化就是把 12345676,12345677,12345678这三个变量对应成1,2,3
这样数组只需要开 int fa[4],这就是离散化的必要性。
思路:该题使用并查集动态维护。起初,所有变量各自构成一个集合;对于每条“相等”的约束条件,合并它约束的两个变量所在的集合。最后扫描所有“不等”的约束条件。若存在一条“不等”的约束条件,它的两个变量处于同一个集合内,则不可能满足。若不存在这样的“不等”约束,则全部条件可以满足。
总结:并查集能在一张无向图中维护节点之间的连通性,这是它的基本用途之一。实际上,并查集擅长动态维护许多具有传递性的关系。所谓传递性,就是A与B有关系,B与C有关系,那么A与C也有某种确定的关系。
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
using namespace std;
pair<int,int>unequal[N];
int n,num,sum,x[N],y[N],op[N],father[N<<1],d[N<<1];
void discrete(int sum)
{
sort(d+1,d+sum+1);
int m=unique(d+1,d+sum+1)-d-1; //使用stl函数unique去重
for(int i=1;i<=n;++i)
{
x[i]=lower_bound(d+1,d+m+1,x[i])-d;
y[i]=lower_bound(d+1,d+m+1,y[i])-d;
}
}
int find(int x)
{
if(father[x]==x) return x;
return father[x]=find(father[x]);
}
void Merge(int x,int y)
{
x=find(x),y=find(y);
if(x!=y) father[x]=y;
}
int main()
{
int t,i;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
int num=0,sum=0;
for(i=1;i<=(n<<1);++i) //n组数据,每组数据两个变量,初始一个变量单独一个集合,所以n<<1乘以2
father[i]=i;
for(i=1;i<=n;++i)
{
scanf("%d%d%d",&x[i],&y[i],&op[i]);
d[++sum]=x[i],d[++sum]=y[i];
}
discrete(sum);
for(i=1;i<=n;++i)
{
if(op[i]==1) Merge(x[i],y[i]);
else unequal[++num]=make_pair(x[i],y[i]);//相等合并,不相等的这对变量放到 pair这个类型数组中,可以存放一对数据
}
for(i=1;i<=num;++i)
{
int x=find(unequal[i].first);
int y=find(unequal[i].second);
if(x==y){puts("NO");goto end;}
}
puts("YES");end:;
}
return 0;
}
//例如x1=x2,那么就把x1 x2放在一个集合中 x1=x3 把x1 x3放到一个集合中,这样x1 x2 x3都在一个集合中
//而第三个条件是x1!=x3 这个条件就不可能成立, 因为上两个条件已经推出x1=x3 这三个条件不能同时成立
//结果输出 NO