第十届蓝桥杯省赛C++A/研究生组 修改数组(暴力及优化）

给定一个长度为 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

//暴力模拟
#include<bits/stdc++.h>
using namespace std;
int main(){
    unordered_set<int>s;
    int n;
    cin>>n;
    vector<int>a(n);
    for(int i=0;i<n;i++)cin>>a[i];
    
    for(int i=0;i<n;i++){
        if(!s.count(a[i])){
            cout<<a[i]<<' ';
            s.insert(a[i]);
        }
        else{
            int t=a[i];
            while(s.count(t)){
                t++;
            }
            cout<<t<<' ';
            s.insert(t);
        }
    }
    return 0;
}

//并查集做法
#include <cstdio>

const int N = 1000010;

int p[N];

int find(int x)
{
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main()
{
    int n;
    scanf("%d", &n);

    for (int i = 1; i < N; i ++ ) p[i] = i;
    for (int i = 1; i <= n; i ++ )
    {
        int x;
        scanf("%d", &x);
        x = find(x);
        printf("%d ", x);
        p[x] = x + 1;
    }

    return 0;
}