力扣 1331. 数组序号转换

题目

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

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

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

示例

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/rank-transform-of-an-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：排序+哈希表

Java实现

class Solution {
    public int[] arrayRankTransform(int[] arr) {
        int n = arr.length;

        int[] arr2 = new int[n];
        arr2 = arr.clone();
        Arrays.sort(arr2);

        Map<Integer, Integer> map = new HashMap<>();
        int flag = 1;
        for (int i = 0; i < n; i++) {
            if (!map.containsKey(arr2[i])) map.put(arr2[i], flag++);
        }
        
        for (int i = 0; i < n; i++) {
            arr[i] = map.get(arr[i]);
        }
        return arr;
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20