题目描述
原题链接
给定一个长度为 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
对于 80% 的评测用例,1 ≤ N ≤ 10000。
对于所有评测用例,1 ≤ N ≤ 100000,1 ≤ Ai ≤ 1000000。
输出
输出N个整数,依次是最终的A1,A2,··· ,AN。
样例
输入
5
2 1 1 3 4
输出
2 1 3 4 5
思路
- 一个数组fa用来存储每个数的父亲,一个数组vis用来记录数字是否出现过。
- 并查集:调整集合,另每个点存的父亲结点都是该联通块的根,即若新输入的数在该集合内,应该被加到该集合中的根再+1。
- 路径压缩:超时 - 递归中需要注意的地方。
代码
#include
#include
using namespace std;int n,x,fa[100010], vis[1100010]; int getfather(int x){if(x == fa[x]){ return x;}fa[x] = getfather(fa[x]); return fa[x];
}int main(){scanf("%d", &n);for(int i &#61; 0; i < 100010; i&#43;&#43;) fa[i] &#61; i; for(int i &#61; 0; i < n; i&#43;&#43;){scanf("%d", &x);if(vis[x]) x &#61; getfather(x)&#43;1; vis[x] &#61; 1; printf("%d ", x); if(x !&#61; 1 && vis[x-1]) fa[x-1] &#61; x; if(vis[x&#43;1]) fa[x] &#61; x&#43;1; }printf("\n");return 0;
}