1.暴力思路
#include <iostream>
#include <bits/stdc++.h>
using namespace std;
#define N 1000005
int exist[N]={0};
int main()
{
// 请在此输入您的代码
int n;
cin>>n;
vector<int> arr(n,0);
for(int i=0;i<n;++i)
{
int x;
cin>>x;
while(exist[x]==1) ++x;
exist[x]=1;
arr.push_back(x);
}
for(int &x:arr)
{
cout<<x<<" ";
}
return 0;
}
2.并查集
#include<bits/stdc++.h>
using namespace std;
const int N=2*1e6+5;
vector<int> father(N,-1);
int find_father(int x){
while(father[x]!=-1){
x=father[x];
}
return x;
}
void Union(int a,int b){
int roota=find_father(a);
int rootb=find_father(b);
if(roota!=rootb)
father[roota]=rootb;
}
int main()
{
int n;
cin>>n;
vector<int> arr(n,0);
for(int i=0;i<n;i++){
int x;
cin>>x;
arr[i]=find_father(x);
Union(arr[i],arr[i]+1);
}
for(int i=0;i<n;i++){
cout<<arr[i]<<" ";
}
return 0;
}
不过都超过时间限制了,可能还要优化一下并查集啥的
例如在查找的过程中直接把父节点指向最后的那个:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int Max = 1000010;
int f[Max];
int N;
int getf(int n)
{
if(f[n] == 0)
return n;
return f[n] = getf(f[n]+1);
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
cin >> N;
int d;
for(int i = 1; i <= N; ++i)
{
cin >> d;
int fd = getf(d);
cout << fd << " ";
f[d] = fd;
f[fd] = fd; // 这句不能漏掉,否则不能跳过已经使用过的数字
}
return 0;
}
这样就AC了,感谢众大佬教我写题~