蓝桥杯真题：修改数组

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了，感谢众大佬教我写题~