python力扣刷题记录——1331. 数组序号转换

题目:
1331. 数组序号转换

给你一个整数数组 arr ,请你将数组中的每个元素替换为它们排序后的序号。 序号代表了一个元素有多大。序号编号的规则如下: 序号从 1
开始编号。 一个元素越大,那么序号越大。如果两个元素相等,那么它们的序号相同。 每个数字的序号都应该尽可能地小。
python力扣刷题记录——1331. 数组序号转换_第1张图片

方法一:

执行用时:344 ms
内存消耗:34.7 MB

思想:先去重,再排序,再哈希查找。字典以去重列表的值作为键,排序号的列表索引作为键。

class Solution:
    def arrayRankTransform(self, arr: List[int]) -> List[int]:
        new_arr = sorted(set(arr))
        dict1 = {
     }
        for k, v in enumerate(new_arr, 1):
            dict1[v] = k
        arr = [dict1[i] for i in arr]
        return arr

补充:
enumerate()用法
用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出数据和数据下标,一般用在 for 循环当中。
enumerate(sequence, [start=0])

letters = ['a', 'b', 'c', 'd', 'e']
for i, letter in enumerate(letters, start=1):
    print(i, letter)
=======结果=======
1 a
2 b
3 c
4 d
5 e

你可能感兴趣的:(数据结构与算法,数据结构)