修改数组

修改数组

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

数据范围
1≤N≤105,
1≤Ai≤106
输入样例:

5
2 1 1 3 4

输出样例:

2 1 3 4 5

代码:

#include<iostream>
#include<algorithm>
using namespace std;
const int N=1000010;
int n;
int p[N];
int find(int x){
    
    
    if(x!=p[x])
        p[x]=find(p[x]);
    return p[x];
}
int main(){
    
    
    cin >> n;
    for(int i=0;i<N;i++)
        p[i]=i;
    for(int i=0;i<n;i++){
    
    
        int t;
        scanf("%d",&t);
        t=find(t);
        if(i!=0)
            printf(" ");
        printf("%d",t);
        p[t]=t+1;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/zhongxinyun/article/details/114270663