并查集

1.蓝桥幼儿园 - 蓝桥云课 (lanqiao.cn)

#include <bits/stdc++.h>
using namespace std;
const int N=2e5+7;
int n,m;
int p[N];
int find(int x){
  if(p[x]==x)return x;
  else return find(p[x]);
}
int main(){
  cin>>n>>m;
  for(int i=1;i<=n;i++)p[i]=i;
  for(int i=1;i<=m;i++){
    int op;int a,b;cin>>op>>a>>b;
    if(op==1){//插入算法
      int pa=find(a);
      int pb=find(b);
      if(pa==pb)continue;//对比的话要去对比find函数
      else p[pa]=pb;//合并的话用手合并
    }
    else cout<<(find(p[a])==find(p[b])?"YES":"NO")<<'\n';
  }
  return 0;
}

 15 届蓝桥杯 14 天省赛冲刺营 1 期 - 合根植物 - 蓝桥云课 (lanqiao.cn)

#include <bits/stdc++.h>
using namespace std;
typedef pair<int,int> PII;
const int N = 1e6+7;
int p[N];//记录父根
int find(int x){
    if(p[x]!=x)p[x]=find(p[x]);//如果你不是父根就去找你的父根去
    return p[x];
}
int num;
int main() {
    int n,m;cin>>n>>m>>num;
    for(int i=1;i<=n*m;i++)p[i]=i;//父根指向自己
    for(int i=1;i<=num;i++){
      int a,b;cin>>a>>b;
      int pa=find(a);
      int pb=find(b);
      if(pa!=pb){
        p[pa]=pb;
      }
    }
    int ans=0;
    for(int i=1;i<=n*m;i++){
      if(p[i]==i)ans++;//根节点的他特征就是指向自己
    }
    cout<<ans;
}

 1.修改数组 - 蓝桥云课 (lanqiao.cn)

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N=1e6+7;
int p[N];//维护一下下次遇见这个数应该输出什么
int find(int x){
  if(p[x]==x)return x;
  else return p[x]=find(p[x]);//如果不是祖先就往前找 并且合并路径
}
signed main(){
  int n;cin>>n;
  for(int i=1;i<=N;i++)p[i]=i;
  for(int i=1;i<=n;i++){
    int x;cin>>x;
    x=find(x);//找到它最前面的祖先
    cout<<find(x)<<' ';
    p[x]=x+1;//最前面的祖先使用了 就继续往更前面迭代
  }
  return 0;
}