修正重复数

给定一个长度为 N 的数组 A = [A₁, A₂, …, Aₙ]，数组中可能有重复出现的整数。现在小猴子要按以下方法将其修改为没有重复整数的数组。小猴子会依次修改 A₂、A₃、…、Aₙ。当修改 Aᵢ时，小明会检查 Aᵢ是否在 A₁～Aᵢ₋₁ 中出现过。如果出现过，则小猴子会给 Aᵢ加上 1；如果新的 Aᵢ仍在之前出现过，小明会持续给 Aᵢ加 1，直到 Aᵢ没有在 A₁～Aᵢ₋₁ 中出现过。当 Aₙ也经过上述修改之后，显然 A 数组中就没有重复的整数。给定一个数组，按规则输出新的数组。

 

用并查集查询。

#include <bits/stdc++.h>

using namespace std;

const int N = 1000002;

int A[N];

int s[N]; //并查集

int find_set(int x){ //用“路径压缩”优化的

    if(x != s[x]) s[x] = find_set(s[x]); //路径压缩

    return s[x];

}

int main(){

    for(int i = 1;i < N;i++) s[i] = i; //并查集的初始化

    int n;

    scanf("%d", &n);

    for(int i = 1;i <= n;i++){

        scanf("%d", &A[i]);

        int root = find_set(A[i]); //查询到并查集的根

        A[i] = root;

        s[root] = find_set(root + 1); //加1

    }

    for(int i = 1;i <= n;i++) printf("%d ", A[i]);

    return 0;

}