【題解】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;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章