【算法】并查集

[蓝桥杯2019初赛]修改数组

题目描述

给定一个长度为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

首先就想到了常规的bool数组，循环+1的方法，然后超时了，得到分数为80。
所以就用了一个数据结构——并查集。
在计算机科学中，并查集是一种树型的数据结构，用于处理一些不交集（Disjoint Sets）的合并及查询问题。
在这个问题里，我们只要实现查找功能就好了。
AC代码并不复杂。如下：

#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 << " ";
    }
}