蓝桥杯真题——修改数组（单链表式并查集）

修改数组

题目描述

给定一个长度为 N 的数组 A=[A1,A2,⋅⋅⋅AN]，数组中有可能有重复出现的整数。

现在小明要按以下方法将其修改为没有重复整数的数组。

小明会依次修改 A2,A3,⋅⋅⋅,AN。

当修改 Ai 时，小明会检查 Ai 是否在 A1∼Ai−1 中出现过。

如果出现过，则小明会给 Ai 加上 1；如果新的 Ai 仍在之前出现过，小明会持续给 Ai 加 1，直到 Ai 没有在 A1∼Ai−1 中出现过。

当 AN 也经过上述修改之后，显然 A 数组中就没有重复的整数了。

现在给定初始的 A 数组，请你计算出最终的 A 数组。

输入格式
第一行包含一个整数 N。

第二行包含 N 个整数 A1,A2,⋅⋅⋅,AN。

输出格式
输出 N 个整数，依次是最终的 A1,A2,⋅⋅⋅,AN。

数据范围
1≤N≤105,
1≤Ai≤106
输入样例：

5
2 1 1 3 4
输出样例：
2 1 3 4 5

解题思路

我本来想暴力解，开了两个数组，一个用来存原数组，一个用来标记某个数是否被用过，对于原数组的每一个数，判断该数是否出现过，如果用过就判断这个数加一是否用过，一直循环，直到找到一个没有被用过的数，但是发现超时了。
单链表的并查集是第一次用，所以记录一下
并查集大家都知道，就像一棵树，但是我们用的这个并查集就像一个单链表，初始p[i]=i，表示该结点未被使用，否则p[i]就指向它右边第一个未被使用的数（因为我们用了路径压缩，如果不用，其实跟我的暴力解法一样会超时）。每次找到p[x]就是找到了x改变之后的值，然后将p[x]的值改为x+1表示该值已被使用（出现过），指向下一个点。

代码实现

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1100010;//注意n的取值
int p[N];
int find(int x){
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
int main()
{
    int n;
    cin>>n;
    for(int i=0;i<N;i++) p[i]=i;//初始化，每个结点指向自己表示未使用过
    for(int i=1;i<=n;i++){
        int x;
        scanf("%d",&x);
        if(p[x]!=x) x=find(x);//找到x之后第一个未被使用的数
        printf("%d ",x);
        p[x]=x+1;//标记x已使用，修改指向
    }
    cout<<endl;
    return 0;
}