1331. 数组序号转换
题目:给你一个整数数组 arr ,请你将数组中的每个元素替换为它们排序后的序号。
链接 https://leetcode.cn/problems/rank-transform-of-an-array/
个人思路
- 首先用一个数组保存排序完的原数组,然后用一个哈希表保存各元素的序号,最后将原属组的元素替换为序号后返回。
class Solution:
def arrayRankTransform(self, arr: List[int]) -> List[int]:
#字典下标从1开始,比如原数组是:[10, 20, 30, 40],则ranks是{10: 1, 20: 2, 30: 3, 40: 4}
ranks = {v: i for i, v in enumerate(sorted(set(arr)), 1)}
#v是元素值,ranks[v]是索引下标
return [ranks[v] for v in arr]