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 <= 10^5
  • -10^9 <= arr[i] <= 10^9

解题思路

思路非常简单,首先对arr中的数字排序,可以得到每个数组的序号。然后遍历arr,输出对应的序号即可。

class Solution:
    def arrayRankTransform(self, arr: List[int]) -> List[int]:
        data = sorted(list(set(arr)))

        st = collections.defaultdict(int)
        for i, v in enumerate(data):
            st[v] = i + 1
        return [st[i] for i in arr]

更简洁的写法:

class Solution:
    def arrayRankTransform(self, arr: List[int]) -> List[int]:
        return map({a: i + 1 for i, a in enumerate(sorted(set(arr)))}.get, arr)

reference:

https://leetcode.com/problems/rank-transform-of-an-array/discuss/489753/C%2B%2BPython-HashMap

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题,希望大家指出!!!

你可能感兴趣的:(leetcode解题指南,Problems)