携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第1天，点击查看活动详情

刷题的日常

一天一题，保持脑子清爽

数组序号转换

来自leetcode的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]

理解题意

由题意我们可以将条件整理如下：

• 题目会给定我们一个数组，要求我们返回每个元素匹配的排名
• 数组中的元素可能会重复
• 排名从1开始，重复的元素排名应该相同

做题思路

最直接的想法当然是进行排序，然后进行排名，因为有重复的数据，所以需要注意下不能重复统计。原数组的位置不能改变，因为返回的是原数组排序之后的排名，那么可以用个map来记录出现索引。流程如下：

• 题目中提示可能没有元素，那么没有元素的用例可以直接返回
• 因为需要排序，又需要用map结构，很容易就想到TreeMap来记录索引出现的位置
• 将数据加入TreeMap之后，就已经将数据排号顺序了，我们只需要便利Map中的索引
• 遍历的过程就已经将数据排好序了
• 最后将索引出现的位置修改为排名直接返回就可以了

代码实现

public class Solution {
    public int[] arrayRankTransform(int... arr) {
        if (arr.length == 0) {
            return arr;
        }
        Map<Integer, List<Integer>> idxMap = new TreeMap<>();
        for (int i = 0; i < arr.length; i++) {
            List<Integer> idx = idxMap.computeIfAbsent(arr[i], o -> new ArrayList<>());
            idx.add(i);
        }
        int cnt = 1;
        for (Map.Entry<Integer, List<Integer>> entry : idxMap.entrySet()) {
            for (Integer idx : entry.getValue()) {
                arr[idx] = cnt;
            }
            cnt++;
        }
        return arr;
    }
}