蓝桥杯-修改数组-python

简介: 蓝桥杯-修改数组-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 。

【输出格式】

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

【样例输入】

1. 5
2. 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,即我们所求的值,这一步保证了下一次我们查询的正确性。

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


代码


1. N=1000000
2. s=[]  #并查集
3. 
4. #寻找根节点并简化路径
5. def find(x):
6. if x!=s[x]:
7.         s[x]=find(s[x])
8. return s[x]
9. 
10. n=int(input())
11. A=input().split()
12. 
13. #初始化
14. for i in range(N):
15.     s.append(i)
16. 
17. for i in range(n):
18.     root=find(int(A[i]))
19.     A[i]=root  #A[i]等于根节点 
20.     s[root]=root+1 #属于这个集合的元素+1,为下一次寻找做准备
21. for i in A:
22.     print(i,end=' ')
目录
相关文章
|
3月前
|
Python
蓝桥杯练习题(一):Python组之入门训练题
这篇文章是关于蓝桥杯Python组的入门训练题,包括Fibonacci数列、圆的面积、序列求和和A+B问题的具体代码实现和样例输出。
149 0
|
3月前
|
存储 机器学习/深度学习 算法
蓝桥杯练习题(三):Python组之算法训练提高综合五十题
蓝桥杯Python编程练习题的集合,涵盖了从基础到提高的多个算法题目及其解答。
130 3
蓝桥杯练习题(三):Python组之算法训练提高综合五十题
|
3月前
|
机器学习/深度学习 并行计算 大数据
【Python篇】NumPy完整指南(上篇):掌握数组、矩阵与高效计算的核心技巧2
【Python篇】NumPy完整指南(上篇):掌握数组、矩阵与高效计算的核心技巧
102 10
|
3月前
|
人工智能 Python
蓝桥杯练习题(四):Python组之历届试题三十题
关于蓝桥杯Python组历届试题的三十个练习题的总结,包括题目描述、输入输出格式、样例输入输出以及部分题目的解题思路和代码实现。
53 0
蓝桥杯练习题(四):Python组之历届试题三十题
|
3月前
|
索引 Python
【Python篇】NumPy完整指南(上篇):掌握数组、矩阵与高效计算的核心技巧1
【Python篇】NumPy完整指南(上篇):掌握数组、矩阵与高效计算的核心技巧
131 4
|
3月前
|
存储 机器学习/深度学习 算法
蓝桥杯练习题(二):Python组之基础练习三十题
蓝桥杯Python编程练习题的集合,包含了三十个不同难度的编程题目,覆盖了基础语法、数据结构和算法等领域。
54 0
|
5月前
|
存储 数据处理 索引
如何删除 Python 数组中的值?
【8月更文挑战第29天】
210 8
|
5月前
|
索引 Python
向 Python 数组添加值
【8月更文挑战第29天】
62 8
|
8月前
|
人工智能 算法 Java
第十四届蓝桥杯集训——练习解题阶段(无序阶段)-ALGO-1005 数字游戏
第十四届蓝桥杯集训——练习解题阶段(无序阶段)-ALGO-1005 数字游戏
116 0
|
8月前
|
Java C语言 C++
第十四届蓝桥杯集训——练习解题阶段(无序阶段)-ALGO-1000 kAc给糖果你吃
第十四届蓝桥杯集训——练习解题阶段(无序阶段)-ALGO-1000 kAc给糖果你吃
88 0