完成数据结构pta实验:
7-3 修改数组(蓝桥杯)
分数 20
作者 liudan
单位 石家庄铁道大学
给定一个长度为 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。
其中,1≤N≤105,1≤Ai≤106。若输入数据不合理,则输出:error
输出格式:
输出 N 个整数,依次是最终的 A1,A2,⋅⋅⋅,AN。
输入样例:
在这里给出一组输入。例如:
5
2 1 1 3 4
输出样例:
在这里给出相应的输出。例如:
2 1 3 4 5
点击查看代码
#include<iostream>
#include<vector>
#include<unordered_set>
using namespace std;
int main(){
int N;
cin>>N;
if(N<1||N>100000){
cout<<"error"<<endl;
return 0;
}
vector<int> A(N);
unordered_set<int> B;
for(int i = 0;i<N;i++){
cin>>A[i];
if(A[i]<1 || A[i]>1000000){
cout<<"error"<<endl;
return 0;
}
}
B.insert(A[0]);
for(int i = 1;i<N;i++){
while(B.count(A[i])){
A[i]++;
}
B.insert(A[i]);
}
for(int i = 0;i<N;i++){
cout<<A[i];
cout<<" ";
}
cout<<endl;
return 0;
}