題意:有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;
}