第十屆藍橋杯省賽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];
	}
}