在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。
考虑一个约束满足问题的简化版本:假设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”表示不可被满足。
数据范围
1≤n≤1000000
1≤i,j≤1000000000
输入样例:
2
2
1 2 1
1 2 0
2
1 2 1
2 1 1
输出样例:
NO
YES
**
这道题主要应用并查集来解决问题
并查集(Disjoint-Set)是一一种可以动态维护若千个不重叠的集合,并支持合并与查询的数据结构。详细地说,并查集包括如下两个基本操作:
1 Get,查询一个元素属于哪一一个集合。
2 Merge,把两个集合合并成一个大集合。
为了具体实现并查集这种数据结构,我们首先需要定义集合的表示方法。在并查集中,我们采用“代表元”法,即为每个集合选择一个固定的元素,作为整个集合的代表
1:并查集的存储
使用一个数组fa保存父节点
int fa[]
2:并查集的初始化
设有n个元素,起初所有元素各自构成个独立的集合,即有n个1个点的数。
for(int i=1;i<=n;i++)fa[i]=i;
3.并查集的Get操作
若x是树根,则x就是集合代表,否则递归访问fa[x]直至根节点。
int get(int x) {
if (x == fa[x]) return X;
return fa[x] = get(fa[x]); // 路径压缩,fa 直接赋值为代表元素
4.并查集的Merge操作
合并元素x和元素y所在的集合,等价于让x的树根作为? y的树根的子节点。void merge(int X, int y) {
fa[get(x)] = get(y);
在这道题中一开始所有的变量各自为一个集合,如果每一条都相等就合并两个变量所在的集合。
如果存在一个不等的约数条件,但是他们却存在相等关系,就不被满足
**
#include <bits/stdc++.h>
using namespace std;
const int N=1000000*2+10;
int a[N][3],n,m,i,j,cnt,t,fa[N],pan[N]; //fa表示一个查并集,pan存储的是在输入的每组数中,最后一位的值
unordered_map<int ,int> c;//因为数的范围可能会较大,所以可以将它先映射到这里
int get(int x)
{
return fa[x]==x?x:fa[x]=get(fa[x]);//寻找他的根
}
void merge(int x,int y)
{
fa[get(x)]=get(y);//合并
}
int main()
{
cin>>t;
while(t--)
{
cin>>n;
n*=3;//因为每组第三个数所以要乘以三
for(int i=1;i<=n;i++)
{
int x;
cin>>x;
if(i%3)//判断是第几个数。
{
if(c[x]==0)//判断这个数有没有出现过
c[x]=(++cnt);
a[i/3+1][i%3]=c[x];//i/3+1表示的是第几组。i%3表示的是第几个数
fa[c[x]]=c[x];将它存入并查集中
}
else
pan[i/3]=x;
}
int ok=0;
for(int i=1;i<=n/3;i++)//开始遍历每组数
{
if(pan[i])
merge(a[i][1],a[i][2]);//如果要是1,就合并他们
}
for(int i=1;i<=n/3;i++)
if(!pan[i])
{
if(get(a[i][1])==get(a[i][2]))//如果是本来不应该相等的数,相等了,就返回NO
{
ok=true;
cout<<"NO";break;
}
}
if(!ok)
cout<<"YES";
cout<<endl;
}
}