当前位置 : 主页 > 编程语言 > python >

1331. 数组序号转换

来源:互联网 收集:自由互联 发布时间:2022-08-10
给你一个整数数组 arr ,请你将数组中的每个元素替换为它们排序后的序号。序号代表了一个元素有多大。序号编号的规则如下: 序号从 1 开始编号。 一个元素越大,那么序号越大。

给你一个整数数组 arr ,请你将数组中的每个元素替换为它们排序后的序号。序号代表了一个元素有多大。序号编号的规则如下:

  • 序号从 1 开始编号。
  • 一个元素越大,那么序号越大。如果两个元素相等,那么它们的序号相同。
  • 每个数字的序号都应该尽可能地小。

    输入:arr = [40,10,20,30]输出:[4,1,2,3]解释:40 是最大的元素。 10 是最小的元素。 20 是第二小的数字。 30 是第三小的数字。

解:去重排序后,使用字典记录每个数字的序号。

class Solution: def arrayRankTransform(self, arr: List[int]) -> List[int]: sorted_arr = sorted(set(arr)) d = {x:i for i,x in enumerate(sorted_arr,1)} return [d[x] for x in arr]

有两个小细节,第一个细节是set也可以用sorted 函数排序。sorted函数的说明:根据 iterable 中的项返回一个新的已排序列表。只要是可迭代的对象都可以使用sorted排序,返回结果是一个新的排序列表。

第二个细节是enumerate可以指定序号开始的值,enumerate(sorted_arr,1) 这里就是从1开始计数。

上一篇:Python入门之——线程threading(Thread类)简介
下一篇:没有了
网友评论