Leetcode 1331.数组序号转换

1 题目描述（Leetcode题目链接）

  给你一个整数数组 arr ，请你将数组中的每个元素替换为它们排序后的序号。

序号代表了一个元素有多大。序号编号的规则如下：

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

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

输入：arr = [100,100,100]
输出：[1,1,1]
解释：所有元素有相同的序号。

提示：

• $0<=arr.length<=10^5$
• $-10^9<=arr[i]<=10^9$

2 题解

  先对数组去重，然后排序，用一个字典记录排序后每个值对应的下标。

class Solution:
    def arrayRankTransform(self, arr: List[int]) -> List[int]:
        seq = sorted(set(arr))
        dic = {}
        for index, num in enumerate(seq, 1):
            dic[num] = index
        return [dic[num] for num in arr]