蓝桥杯真题-修改数组

题目

 一、暴力法 

 暴力法只能过三个，其余七个都会超时

n = int(input())
arr = list(map(int,input().split()))

if n == 1:
    print(arr[0])
else:
    for i in range(1,n):
        while(arr[i] in arr[:i]):
            arr[i] += 1
for i in arr:
    print(i,end = " ")

二、并查集

主要的注意点在于并查集需要压缩路径，否则仍然可能超时。

这里的思想就是首先判断这个数是否已经出现过，只需要判断它的父节点是否是它本身：

如果x没有出现那么fa[x]必然是它自己，判断过后需要标注自己已经出现过了，union(x,x+1)

如果x已经出现过即 fa[x] != x,那么只需要找到x的祖宗节点find(x)即可，这里不难发现其实祖宗节点都是还没有出现过的节点可以作为候选。

fa = [i for i in range(10**6)]
def find(x):
    if fa[x] == x:
        return x
    else:
        fa[x] = find(fa[x])
        return fa[x]
def union(a,b):
    a_fa = find(a)
    b_fa = find(b)
    fa[a_fa] = b_fa
n = int(input())
arr = list(map(int,input().split()))
for i in range(n):
    if find(arr[i]) != arr[i]:
        arr[i] = find(arr[i])
    union(arr[i],arr[i]+1)
for i in arr:
    print(i,end = " ")