励志用更少的代码做更高效的表达
题目描述
给定一个长度为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(1<=N<=100000)
第二行包含N个整数A1,A2,…,AN(1<=Ai<=1000000)
输出
输出N个整数,依次是最终的A1,A2,…,AN
样例输入
5
2 1 1 3 4
样例输出
2 1 3 4 5
用常规的数组暴力解法大概能得80%的分。
也可以用并查集来解。只要实现其查找功能即可
代码展示
#include<bits/stdc++.h>
using namespace std;
//并查集
int f[1000001];
int getf(int x) {
if (x == f[x])
return x;
return f[x] = f[getf(f[x])];
}
int main() {
for (int i=1;i<1000001;i++)
f[i] = i;
int n, t;
cin >> n;
while (n--){
cin >> t;
t = getf(t);
cout << t;
f[t] = getf(t + 1);
if (n >= 1)cout << " ";
}
return 0; }