蓝桥杯2019修改数组（运用并查集）

2019修改数组（运用并查集）
代码如下：

//2019修改数组（运用并查集）
#include
#define rep(i,a,b) for(int i=(a);i<=(b);(i++))
using namespace std;
const int maxn=100000+100;
int mp[maxn];
int findmp(int x)//沿着上级寻找掌门人，也就是可以插入的数
{
int tx=x;
while(mp[x]!=x)//如果指向自己，说明是掌门人
    x=mp[x];
int res=x;
while(mp[x]!=x)//回到起点，把沿途的人都直接指向掌门人，压缩路径
    {
    int temp=mp[x];
    mp[x]=res;
    x=temp;
}
return res;

}

int main(){
    for(int i=0;i<=maxn;i++)//初始化并查集，各自为盟
        mp[i]=i;
vector<int> result;
int n;
cin>>n;
rep(i,1,n){
int temp;
cin>>temp;
int inser=findmp(temp);
result.push_back(inser);
mp[inser]=inser+1;//插入以后，不在是掌门人，指向下一个数，连接到新的门派

}
for(vector<int>::iterator iter=result.begin();iter!=result.end();iter++)
    cout<<*iter<<" ";
    cout<<endl;

return 0;
}