第十届蓝桥杯——修改数组

【问题描述】

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

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

小明会依次修改 A₂, A₃, ⋅⋅⋅, A_N。

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

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

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

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

【输入格式】
第一行包含一个整数 N。
第二行包含 N 个整数 A₁, A₂, ⋅⋅⋅, A_N。

【输出格式】
输出 N 个整数，依次是最终的 A₁, A₂, ⋅⋅⋅, A_N。

【输入样例】

5

2 1 1 3 4

【输出样例】

2 1 3 4 5

【数据范围】
1 ≤ N ≤ 10⁵
1 ≤ A_i ≤ 10⁶

---

题解一
暴力（会超时）： 两重循环，应该拿不了多少分数。。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1000010;

int n;
int a[N];
bool st[N];

int main()
{
    cin >> n;
    for (int i = 0; i < n; i++) scanf("%d", &a[i]);
        
    st[a[0]] = true;
    for (int i = 1; i < n; i ++)
    {
        if (!st[a[i]]) st[a[i]] = true;
        else
        {
            while (st[a[i]]) a[i] ++;
            st[a[i]] = true;
        }
    }
    
    for (int i = 0; i < n; i++) printf("%d ", a[i]);
    return 0;
}

题解二
并查集：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1100010;

int n;
int a[N];

int find(int x)                                 // 寻找父节点 + 路径压缩
{
    if(a[x] != x) a[x] = find(a[x]);            // 如果不指向父节点
    return a[x];
}

int main()
{
    for (int i = 1; i <= N; i ++) a[i] = i;     // 初始化，自己是自己的父节点
    
    cin >> n;
    for (int i = 1; i <= n; i ++)
    {
        int x;
        scanf("%d", &x);
        x = find(x);                            // 找到 x 指向的父节点
        printf("%d ", x);
        a[x] = x + 1;                           // 将 x 的父节点更新为 x+1
    }
    
    return 0;
}