【LetMeFly】1331.数组序号转换

力扣题目链接：https://leetcode.cn/problems/rank-transform-of-an-array/

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

方法一：sort + 哈希表

首先把原始数组拷贝一份到临时数组中，并对临时数组进行排序

排序后，遍历一遍临时数组，把名次记录下来（用一个变量last来存放上一个值，若这个值与上一个不同就名次+1）到哈希表中

遍历原始数组，把值修改“从哈希表中映射为名次后的值”

• 时间复杂度$O(n\log n)$，其中$n$是数组中元素个数。排序耗时$n\log n$
• 空间复杂度$O(N\log N)$。哈希表、排序和临时数组消耗空间都是$O(n)$

AC代码

C++

class Solution {
public:
    vector<int> arrayRankTransform(vector<int>& arr) {
        vector<int> toSort = arr;
        sort(toSort.begin(), toSort.end());
        unordered_map<int, int> ma;
        int th = 0;
        int last = -1e9 - 1;
        for (int i = 0; i < toSort.size(); i++) {
            if (toSort[i] == last)
                continue;
            ma[toSort[i]] = ++th;
            last = toSort[i];
        }
        for (int& t : arr) {
            t = ma[t];
        }
        return arr;
    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

Python

语法糖真简洁

Python代码 Copy From 力扣官方题解

class Solution:
    def arrayRankTransform(self, arr: List[int]) -> List[int]:
        ranks = {v: i for i, v in enumerate(sorted(set(arr)), 1)}
        return [ranks[v] for v in arr]
1
2
3
4

同步发文于CSDN，原创不易，转载请附上原文链接哦~
Tisfy：https://letmefly.blog.csdn.net/article/details/126030218