LeetCode-Python-5155. 数组序号转换(排序 + 哈希表)

给你一个整数数组 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]

思路:

直接排序即可,注意如果两个元素相等,那么它们的序号相同。

时间复杂度:O(NlogN)

空间复杂度::O(N)

class Solution(object):
    def arrayRankTransform(self, arr):
        """
        :type arr: List[int]
        :rtype: List[int]
        """
        l = sorted(arr)
        dic, rank = {}, 0
        pre_num = None
        for i, num in enumerate(l):
            if pre_num != num:
                rank += 1
                dic[num] = rank
                pre_num = num
        
        res = []
        for num in arr:
            res.append(dic[num])
        return res

 

发布了722 篇原创文章 · 获赞 106 · 访问量 19万+

猜你喜欢

转载自blog.csdn.net/qq_32424059/article/details/104085430