【并查集】修改数组【蓝桥杯】

题目

修改数组 - 蓝桥云课

---

题目描述

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

现在小明要按以下方法将其修改为没有重复整数的数组。小明会依次修改 $A_2,A_3,··· ,A_N$ 。

当修改 $A_i$ 时，小明会检查 $A_i$ 是否在 $A_1$ ∼ $A_i−1$ 中出现过。如果出现过，则小明会给 $A_i$ 加上 1 ；如果新的 $A_i$ 仍在之前出现过，小明会持续给 $A_i$ 加 1 ，直 到 $A_i$ 没有在 $A_1$ ∼ $A_i−1$ 中出现过。

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

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

输入描述

第一行包含一个整数 $N$ 。

第二行包含 $N$ 个整数 $A_1,A_2,··· ,A_N$ 。

其中， $1 \leq N \leq 10^5，1 \leq A_i \leq 10^6$ 。

输出描述

输出 $N$ 个整数，依次是最终的 $A_1,A_2,··· ,A_N$ 。

输入输出样例

示例

输入

5
2 1 1 3 4

输出

2 1 3 4 5

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

解题

方法一：单链表式的并查集

思路

代码

import java.util.*;
import java.io.*;

public class Main {
    static final int N = (int) 1e6 + 10;
    static int[] roots = new int[N];
    static {
        for (int i = 1; i < N; ++i) roots[i] = i;
    }

    static int find(int x) {
        return roots[x] == x ? x : (roots[x] = find(roots[x]));
    }

    public static void main(String[] args) throws IOException {
        StreamTokenizer in = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
        in.nextToken(); int n = (int) in.nval;
        for (int i = 0; i < n; ++i) {
            in.nextToken(); int x = (int) in.nval;
            x = find(x); 
            roots[x] = x + 1;
            System.out.print(x + " ");
        }
    }
}