第十届蓝桥杯省赛Java 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。
【样例输入】
5
2 1 1 3 4

【样例输出】
2 1 3 4 5

【评测用例规模与约定】
对于 80% 的评测用例，1 ≤ N ≤ 10000。
对于所有评测用例，1 ≤ N ≤ 100000，1 ≤ Ai ≤ 1000000。

代码如下：

import java.io.*;
import java.util.*;
public class Main {
	static int[] a = new int[100010];
	static int[] f = new int[1000010]; //注意范围
	static int n;
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(new BufferedInputStream(System.in));
		n = sc.nextInt();
		for(int i = 1; i <= n; i++)
			a[i] = sc.nextInt();
		
		for(int i = 1; i <= 1000003; i++) //初始化并查集
			f[i] = i;
		
		for(int i = 1; i <= n; i++) {
			a[i] = find(a[i]);
			f[a[i]] = find(a[i] + 1);
		}
		
		for(int i = 1; i <= n; i++)
			System.out.print(a[i] + " ");
	}
	
	public static int find(int x) {
		if(f[x] != x)
			f[x] = find(f[x]);
		return f[x];
	}
}