题目描述
给定一个长度为 N 的数组 A=[A1,A2,⋯AN],数组中有可能有重复出现的整数。
现在小明要按以下方法将其修改为没有重复整数的数组。小明会依次修改 A2,A3,⋯,AN。
当修改 Ai 时,小明会检查 Ai 是否在 A1 ∼ Ai−1 中出现过。如果出现过,则小明会给 Ai 加上 11;如果新的Ai 仍在之前出现过,小明会持续给 Ai 加 11,直到 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
思路:
首先先简化题意,即每一个数前面都不能出现相同的数
第一种做法--set
思路:因为我们已知Ai小于10的六次方,所以我们可以考虑lower_bound函数每次找每个数第一个大于它的数,然后把它从set容器里拿掉
ACcode
#include <iostream>
#include <set>
using namespace std;
int n,x;
int q[100010];
set<int> s;
int main()
{
ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
cin>>n;
for(int i=1;i<=1e6;i++) s.insert(i);初始化set
for(int i=1;i<=n;i++)
{
cin>>x;
auto a=s.lower_bound(x);找到第一个大于等于它的数对应的迭代器
s.erase(*a);删去该数据
cout<<*a<<" ";
}
return 0;
}
第二种做法--并查集
思路:首先先让所有的f(i)=i,即每个人最开始的祖先都是自己,然后就每一次都让轮到那个数的父亲+1,第二次出现的时候就直接用父亲替换掉
ACcode
#include <iostream>
using namespace std;
int n,x;
int q[100010],f[1000010];
int Find(int x)
{
if(f[x]==x) return x;
return f[x]=Find(f[x]);
}
void unionset(int x,int y)
{
f[Find(x)]=Find(y);
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
cin>>n;
for(int i=1;i<=1e6;i++) f[i]=i; 初始化
for(int i=1;i<=n;i++)
{
cin>>x;
x=Find(x);
f[x]=Find(x)+1;
cout<<x<<' ';
}
return 0;
}