1459: [蓝桥杯2019初赛]修改数组

题目描述
给定一个长度为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(1<=N<=100000)
第二行包含N个整数A1,A2,…,AN(1<=Ai<=1000000)
输出
输出N个整数，依次是最终的A1,A2,…,AN
***样例输入 ***
5
2 1 1 3 4
样例输出
2 1 3 4 5

运用并查集

package test_1459;

import java.util.Scanner;

public class Main {

	static int[] a = new int[100010];
	static int[] f = new int[100010];
	static int n;
	static Scanner cin = new Scanner(System.in);

	public static void main(String[] args) {
		// TODO 自动生成的方法存根
		// 初始化f数组为它本身
		for (int i = 1; i <= 100009; i++) {
			f[i] = i;
		}
		n = cin.nextInt();
		for (int i = 1; i <= n; i++) {
			a[i] = cin.nextInt();
			a[i] = find(a[i]);
			f[a[i]] = find(a[i] + 1);
		}

		for (int i = 1; i <= n; i++) {
			System.out.print(a[i] + " ");
		}
	}

	static int find(int x) {
		if (f[x] == x) {
			return x;
		} else {
			f[x] = find(f[x]);
			return f[x];
		}
	}
}