蓝桥杯-修改数组-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=' ')
目录
相关文章
|
12天前
|
存储 Java 数据处理
(numpy)Python做数据处理必备框架!(一):认识numpy;从概念层面开始学习ndarray数组:形状、数组转置、数值范围、矩阵...
Numpy是什么? numpy是Python中科学计算的基础包。 它是一个Python库,提供多维数组对象、各种派生对象(例如掩码数组和矩阵)以及用于对数组进行快速操作的各种方法,包括数学、逻辑、形状操作、排序、选择、I/0 、离散傅里叶变换、基本线性代数、基本统计运算、随机模拟等等。 Numpy能做什么? numpy的部分功能如下: ndarray,一个具有矢量算术运算和复杂广播能力的快速且节省空间的多维数组 用于对整组数据进行快速运算的标准数学函数(无需编写循环)。 用于读写磁盘数据的工具以及用于操作内存映射文件的工具。 线性代数、随机数生成以及傅里叶变换功能。 用于集成由C、C++
202 1
|
Python
蓝桥杯练习题(一):Python组之入门训练题
这篇文章是关于蓝桥杯Python组的入门训练题,包括Fibonacci数列、圆的面积、序列求和和A+B问题的具体代码实现和样例输出。
284 0
|
存储 机器学习/深度学习 算法
蓝桥杯练习题(三):Python组之算法训练提高综合五十题
蓝桥杯Python编程练习题的集合,涵盖了从基础到提高的多个算法题目及其解答。
528 3
蓝桥杯练习题(三):Python组之算法训练提高综合五十题
|
机器学习/深度学习 并行计算 大数据
【Python篇】NumPy完整指南(上篇):掌握数组、矩阵与高效计算的核心技巧2
【Python篇】NumPy完整指南(上篇):掌握数组、矩阵与高效计算的核心技巧
348 10
|
人工智能 Python
蓝桥杯练习题(四):Python组之历届试题三十题
关于蓝桥杯Python组历届试题的三十个练习题的总结,包括题目描述、输入输出格式、样例输入输出以及部分题目的解题思路和代码实现。
333 0
蓝桥杯练习题(四):Python组之历届试题三十题
|
索引 Python
【Python篇】NumPy完整指南(上篇):掌握数组、矩阵与高效计算的核心技巧1
【Python篇】NumPy完整指南(上篇):掌握数组、矩阵与高效计算的核心技巧
381 4
|
存储 数据处理 索引
如何删除 Python 数组中的值?
【8月更文挑战第29天】
516 9
|
存储 机器学习/深度学习 算法
蓝桥杯练习题(二):Python组之基础练习三十题
蓝桥杯Python编程练习题的集合,包含了三十个不同难度的编程题目,覆盖了基础语法、数据结构和算法等领域。
244 0
|
人工智能 Python
【蓝桥杯国赛真题笔记】Python(2)
【蓝桥杯国赛真题笔记】Python
329 0
【蓝桥杯国赛真题笔记】Python(2)
|
移动开发 Shell Python
【蓝桥杯国赛真题笔记】Python(1)
【蓝桥杯国赛真题笔记】Python
263 0
【蓝桥杯国赛真题笔记】Python(1)

推荐镜像

更多