蓝桥杯-修改数组-python

链接：https://www.lanqiao.cn/problems/185/learning/

目录

问题描述

思路

代码

问题描述

给定一个长度为 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 。
【输出格式】
输出 N 个整数，依次是最终的 A1, A2, · · · , AN。
【样例输入】

5
2 1 1 3 4

【样例输出】

2 1 3 4 5

【评测用例规模与约定】
对于 80% 的评测用例，1 ≤ N ≤ 10000。
对于所有评测用例，1 ≤ N ≤ 100000，1 ≤ Ai ≤ 1000000。
时间限制: 1.0s 内存限制: 256.0MB 本题总分：20 分

思路

使用并查集的算法，先寻找A[i]的根节点root，将root加入到答案中，再让s[root]这个集合的值加一，这一步的意义是：如果再次遇到之前遇到的数，可以直接在最终的答案里写入这个数根节点的值+1，即我们所求的值，这一步保证了下一次我们查询的正确性。

这样循环往复，输入值之后就可以在最终的答案里加入正确的值。

代码

N=1000000
s=[]  #并查集

#寻找根节点并简化路径
def find(x):
    if x!=s[x]:
        s[x]=find(s[x])
    return s[x]
 
n=int(input())
A=input().split()
 
#初始化
for i in range(N):
    s.append(i)
    
for i in range(n):
    root=find(int(A[i]))
    A[i]=root  #A[i]等于根节点 
    s[root]=root+1 #属于这个集合的元素+1，为下一次寻找做准备
for i in A:
    print(i,end=' ')