[蓝桥杯 2019 省赛 A组] 修改数组详解(自己解法+并查集)

最新推荐文章于 2024-06-13 19:47:49 发布

RedCap.

最新推荐文章于 2024-06-13 19:47:49 发布

阅读量906

点赞数 27

分类专栏：蓝桥杯之我试文章标签：蓝桥杯 python 算法

本文链接：https://blog.csdn.net/wrj1472583690/article/details/137173007

版权

蓝桥杯之我试专栏收录该内容

20 篇文章 0 订阅

订阅专栏

题目描述

给定一个长度为 $N$ 的数组 $A=[A_1,A_2, \cdots A_N]$ ，数组中有可能有重复出现的整数。

现在小明要按以下方法将其修改为没有重复整数的数组。小明会依次修改 $A_2,A_3, \cdots ,A_N$ 。

当修改 $A_i$ 时，小明会检查 $A_i$ 是否在 $A_1$ ∼ $A_{i-1}$ 中出现过。如果出现过，则小明会给 $A_i$ 加上 $1$ ；如果新的 $A_i$ 仍在之前出现过，小明会持续给 $A_i$ 加 $1$ ，直到 $A_i$ 没有在 $A_1$ ∼ $A_{i-1}$ 中出现过。

当 $A_N$ 也经过上述修改之后，显然 $A$ 数组中就没有重复的整数了。

现在给定初始的 $A$ 数组，请你计算出最终的 $A$ 数组。

输入格式

第一行包含一个整数 $N$ 。

第二行包含 $N$ 个整数 $A_1,A_2, \cdots ,A_N$ 。

输出格式

输出 $N$ 个整数，依次是最终的 $A_1,A_2, \cdots ,A_N$ 。

样例 #1

样例输入 #1

5
2 1 1 3 4

样例输出 #1

2 1 3 4 5

提示

对于 $80\%$ 的评测用例， $\le N \le 10000$ 。

对于所有评测用例， $\le N \le 10^5$ ， $\le A_i \le 10^6$ 。

蓝桥杯 2019 年省赛 A 组 H 题。

1 暴力 40%
直接每次看见不满足后就进行加1直到满足为止

2 暴力优化 100%（看算法标签有并查集加入后感觉没啥用）
为什么暴力太慢？
因为我们重复计算了太多次已经遍历过的状态
所以我加入了在遍历有如果加了超过20次1，就开一个字典记住。
下一次直接跳到遍历过的结尾去再加就好了。
注意p数组要开大一点，不然会RE

n = int(input())

h = [0] + list(map(int,input().split()))

p = [i for i in range(n+1)]

'''
st = [0 for i in range(int(1e6+1))]

for i in range(1,n+1):
    if st[h[i]] == 0:
        st[h[i]] = 1
    else:
        j = h[i]
        while st[j] ==1:
            j+=1
        h[i] = j
        st[j] = 1
    print(h[i],end =' ')
'''

def root(x):
    if x != p[x]:
        p[x] = root(p[x])
    return p[x]

def union(x,y):
    if root(x) != root(y):
        p[root(x)] =root(y)
d = dict()
for i in range(2,n+1):
      if root(h[i]) != root(h[1]):union(h[i] , h[1])
      else:
          j = h[i] ; f =False;
          for l,r in d.items():
              if l<=j and r>=j :
                  j=r; h[i] = l
                  f = True
                  break
          while root(j) == root(h[1]):
              j+=1
          union(h[1],j)
          if j - h[i] >= 15 or f: d[h[i]] = j
          h[i] = j

for ii in h[1:]:
    print(ii,end = ' ')