修改数组 第十届蓝桥杯 C++

题目描述

原题链接
给定一个长度为 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 <iostream>
#include <cstdio>
using namespace std;

int n,x,fa[100010], vis[1100010]; //fa数组存该点的父亲值

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 = 0; i < 100010; i++) fa[i] = i; //初始化
    for(int i = 0; i < n; i++){
        scanf("%d", &x);
        if(vis[x]) x = getfather(x)+1; 
        //该点出现过 找她的父亲根节点（上限最大） 然后加1
        vis[x] = 1; //新点的访问变成1
        printf("%d ", x); //打印结果
        if(x != 1 && vis[x-1])  fa[x-1] = x; 
        //如果该点和前一个数连起了，把前一个数的爸爸改成这个点
        if(vis[x+1])   fa[x] = x+1; //如果和后一个点连起了，则这个点的饿爸爸改成后一个点
    }
    printf("\n");
    return 0;
}