1331. 数组序号转换

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

  • 序号从 1 开始编号。
  • 元素越大,序列号越大。如果两个元素相等,则它们具有相同的序列号。
    [En]

    the larger an element, the larger the serial number. If two elements are equal, they have the same sequence number.*

  • 每个号码的序号应尽可能小。
    [En]

    the sequence number of each number should be as small as possible.*

输入: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开始计数。

Original: https://blog.51cto.com/pigeon/5521368
Author: 一只大鸽子
Title: 1331. 数组序号转换

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/504714/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球