数组序号转换

一起养成写作习惯！这是我参与「掘金日新计划 · 4 月更文挑战」的第20天，点击查看活动详情。

题目描述

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

提示

• 0 <= arr.length <= 10^5
• -1^9 <= arr[i] <= 10^9

排序+计数

我们可以将数据进行排序，以此来得到元素对应的序号。在此基础上，需要注意的地方有原数组的顺序需要保留，同时相同的元素需要一致。

步骤：

1. 先将原数组复制一份出来，然后进行排序
2. 统计不同的数值，以及其排序后的序号
3. 遍历原数组，将当前元素对应的序号取出，更新元素

class Solution {
    public int[] arrayRankTransform(int[] arr) {
        // 边界判断
        int n = arr.length;
        if(0 == n){
            return arr;
        }

        // 复制数值并排序
        int[] clone = arr.clone();
        Arrays.sort(clone);

        Map<Integer, Integer> map = new HashMap<>();
        int count = 1;
        int prev = clone[0];
        map.put(prev, count++);
        
        // 遍历排序后的数组
        for(int c : clone){
            // 将不同的元素及其序号存入结果集
            if(prev != c){
                map.put((prev = c), count++);
            }
        }

        // 遍历原数组，取出序号复制给当前元素
        for(int i = 0; i < n; ++i){
            arr[i] = map.get(arr[i]);
        }

        // 返回结果
        return arr;
    }
}