题意:有n个单词,给定m个关系,每个关系要么表示单词a与单词b相同,要么表示单词a与单词b相反。并且“相同”与“相反”有性质:若a与b相同,b与c相同,则a与c相同(从而单词的相同关系是等价关系);若a与b相反,b与c相反,则a与c相同。按顺序判断这m个关系是否可以成立,若可以成立,则加上这个关系,否则忽略。再给定q个询问,每个询问 查询单词a与单词b的关系(相同、相反或未知)。
分析:相同的单词放在一个集合里,用并查集维护。对每个单词a,定义at[a]表示与a相反的一个单词,初始at[a]=0。对关系分类处理
1.单词a与单词b相同
判断是否可以成立:可以成立当且仅当b不属于a所在的集合的相反集合,即find(b)!=find(at[find(a)])。
若成立加上该关系:将b所在的集合合并至a所在的集合,再将at[b]所在的集合合并至at[a]所在的集合。
2.单词a与单词b相反
判断是否可以成立:可以成立当且仅当b不属于a所在的集合,即find(b)!=find(a)。
若成立加上该关系:将a所在的集合合并至at[b]所在的集合,将b所在的集合合并至at[a]所在的集合。
询问相对容易处理。
代码
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
int n,m,q,pre[maxn],at[maxn];
map<string,int> mp;
int find(int u)
{
if (!u) return 0;
if (pre[u]!=u) pre[u]=find(pre[u]);
return pre[u];
}
int main()
{
std::ios::sync_with_stdio(false);
cin>>n>>m>>q;
for (int i=1;i<=n;i++)
{
string name;
cin>>name;
mp[name]=i;
pre[i]=i;
}
while (m--)
{
int t,u,v,u1,v1;
string a,b;
cin>>t>>a>>b;
u=find(mp[a]);v=find(mp[b]);
u1=find(at[u]);v1=find(at[v]);
if (t==1)
{
if (u1==v) printf("NO\n");
else
{
printf("YES\n");
pre[u]=v;
if (u1&&v1) pre[u1]=v1;
if (!v1) at[v]=u1;
}
}
else
{
if (u==v) printf("NO\n");
else
{
printf("YES\n");
if (u1) pre[v]=u1;
else at[u]=v;
if (v1) pre[u]=v1;
else at[v]=u;
}
}
}
while (q--)
{
string a,b;
cin>>a>>b;
int u=find(mp[a]),v=find(mp[b]);
int u1=find(at[u]),v1=find(at[v]);
if (u==v) printf("1\n");
else
{
if (u1==v||v1==u) printf("2\n");
else printf("3\n");
}
}
return 0;
}