【1331. 数组序号转换】

来源：力扣（LeetCode）

描述：

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

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

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

示例 1：

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

示例 2：

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

示例 3：

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

提示：

• 0 <= arr.length <= 10⁵
• -10⁹ <= arr[i] <= 10⁹

方法：排序 + 哈希表

思路

首先用一个数组保存排序完的原数组，然后用一个哈希表保存各元素的序号，最后将原属组的元素替换为序号后返回。

代码:

class Solution {
public:
    vector<int> arrayRankTransform(vector<int>& arr) {
        vector<int> sortedArr = arr;
        sort(sortedArr.begin(), sortedArr.end());
        unordered_map<int, int> ranks;
        vector<int> ans(arr.size());
        for (auto &a : sortedArr) {
            if (!ranks.count(a)) {
                ranks[a] = ranks.size() + 1;
            }
        }
        for (int i = 0; i < arr.size(); i++) {
            ans[i] = ranks[arr[i]];
        }
        return ans;
    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

执行用时：100 ms, 在所有 C++ 提交中击败了39.83%的用户
内存消耗：38.3 MB,在所有 C++ 提交中击败了46.48%的用户
复杂度分析
时间复杂度：O(n × logn)，其中 n 是输入数组 arr 的长度，排序消耗O(n × logn) 时间。
空间复杂度：O(n)。有序数组和哈希表各消耗O(n) 空间。
arthor：LeetCode-Solution