int superior[MAXN]; int find_root(int son) { // int temp=son; // while(son!=superior[son]) // son=superior [son]; // superior[temp]=son;//路径压缩 // return son; if(son==superior[son]) { return son; }else { superior[son]=find_root(superior[son]);//路径压缩 return superior[son]; } } void union_fa(int x1,int x2) { int x1_superior=find_root(x1); int x2_superior=find_root(x2); if(x1_superior!=x2_superior) superior[x2_superior]=x1_superior; }
//初始化
for(int i=0;i<=n;i++) superior[i]=i;
测试代码
superior[1]=2;
superior[2]=4;
superior[4]=0;
superior[5]=1;
superior[3]=5;
superior[0]=100;
cout<<find_root(3);
结果:100