ACWing 836.統合セット(テンプレートタイトルと検索集)
868 ワード
テーマリンク
一山の数字に対して行って、そして集の基本的な操作を調べます.
一山の数字に対して行って、そして集の基本的な操作を調べます.
#include
using namespace std;
const int N=1e5+5;
int p[N];
int find(int x){
if(p[x]!=x) return p[x]=find(p[x]);
return p[x];// return x
}
int main(){
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i) p[i]=i;
char op[2];
while(m--){
int a,b;
scanf("%s%d%d",&op,&a,&b);
if(op[0]=='M'){
p[find(a)]=find(b);
}else{
if(find(b)==find(a)) puts("Yes");
else puts("No");
}
}
}
アルゴリズムの競技の進級の手引きの上でfind関数はこのように書きます.int find(int x){
if(x==p[x]) return x;
return p[x]=find(p[x]);
}
行を押してこのように書くことができます.int find(int x){return p[x]==x?x:p[x]=find(p[x]);}
記憶化検索もこのように書いています.