hihocoder#1066 : 无间道之并查集

#include<map>
#include<vector>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<stack>
#include<queue>
#include<set>
#define inf 0x3f3f3f3f
#define mem(a,x) memset(a,x,sizeof(a))

using namespace std;

typedef long long ll;
typedef pair<int,int> pii;

inline int in()
{
int res=0;char c;
while((c=getchar())<'0' || c>'9');
while(c>='0' && c<='9')res=res*10+c-'0',c=getchar();
return res;
}
map<string,string> mp;
string find(string x)
{
if(mp[x]=="")return x;    //根节点都是为空的
return mp[x]=find(mp[x]);
}
int main()
{
int n=in();
string t1,t2;
for(int i=0;i<n;i++)
{
int q=in();
cin>>t1>>t2;
if(q==0)
{
t1=find(t1),t2=find(t2);
if(t1!=t2)
{
mp[t1]=t2;
}
}
else
{
t1=find(t1),t2=find(t2);
if(t1!=t2)
{
puts("no");
}
else  puts("yes");
}

}
return 0;
}