蓝桥杯真题—修改数组（并查集）

题目描述

给定一个长度为 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。
其中 1 ≤ N ≤ 10⁵，1 ≤ A_i ≤ 10⁶。

输出描述

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

解题思路

初始化并查集，表示数字从未使用过，若一个数被使用，则将其父节点指向第一个未使用并比该数大的节点，具体实现见代码。

代码

n = int(input())
x = list(map(int, input().split()))
f = [i for i in range(int(1e6+1))]
def find(x):
    if f[x] != x:
        f[x] = find(f[x])
    return f[x]
for i in range(n):
    x[i] = find(x[i])
    f[x[i]] = find(x[i] + 1)
    print(x[i], end=' ')