【题解】codeforces766D Mahmoud and a Dictionary

题目链接

题意：有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;
}