Python:修改数组

题目描述

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

其中，1 ≤ N ≤ 10^5，1 ≤ Ai​ ≤ 10^6。

输出描述

输出 N 个整数，依次是最终的 A1​,A2​,⋅⋅⋅,AN​。

输入输出样例

示例

输入

5
2 1 1 3 4

输出

2 1 3 4 5

 参考代码：（暴力算法，运行超时）

n=int(input())
a=[int(i) for i in input().split()]
for i in range(1,n):
  while a[i] in a[0:i]:
    a[i]+=1
for i in a:
  print(i,end=' ')   #end=' ' 用于在一行输出

set（）方法 （超时）

n=int(input())
a=[int(i) for i in input().split()]
s=set()         # 创建集合
for i in range(n):
  while a[i] in s:
    a[i]+=1
  s.add(a[i])   # a[i]不在s中，将a[i]添加到集合s内
for i in a:
  print(i,end=" ")

并查集方法（通过）

def find_set(x):  #查找
  if (x!=s[x]):
    s[x]=find_set(s[x])
  return s[x]
N=200000          #比输入范围大就好
s=list(range(N))  #并查集初始化
n=int(input())
a=[int(i) for i in input().split()]
for i in range(n):
  root=find_set(a[i])
  a[i]=root
  s[root]=find_set(root+1) # 加一
for i in a:
  print(i,end=" ")