题目 2301: 修改数组

题目描述:

给定一个长度为 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 数组

代码:

package lanqiao;

import java.math.BigInteger;
import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] count = new int[1000001];
        long[] arr = new long[n];
        for(int i = 0;i < n;i ++)
        {
            int num = sc.nextInt();
            if(count[num] == 0){
                arr[i] = num;
                count[num] += 1;
            }else{
                while(count[num] != 0){
                    count[num] += 1;
                    num += count[num] - 1;
                }
                count[num] += 1;
                arr[i] = num;
            }
            System.out.print(arr[i] + " ");
        }
    }
}