LeetCode每日一题——1331. 数组序号转换

题目

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

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

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

示例

示例 1：

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

示例 2：

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

示例 3：

输入：arr = [37,12,28,9,100,56,80,5,12]
输出：[5,3,4,2,8,6,7,1,3]

提示：

0 <= arr.length <= 105
-109 <= arr[i] <= 109

思路

题目要求重复元素序号相同，这里干脆直接把重复元素删除，使用set集合来做

1. 对给定的数组转化为集合后排序
2. 将排序后的数组值和对应的下标+1存在哈希表中
3. 遍历原数组，取出哈希表中的值即可

题解

class Solution:
    def arrayRankTransform(self, arr: List[int]) -> List[int]:
        # 排序
        temp = sorted(set(arr))
        res = {}
        # 存哈希
        for i in range(len(temp)):
            res[temp[i]] = i + 1
        # 取
        return [res.get(i) for i in arr]