leetcode 之 1331. 数组序号转换

136 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第18天,点击查看活动详情

题目描述

1331. 数组序号转换 - 力扣(LeetCode)

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

思路分析

  1. 拷贝原来的数组,进行排序;这里大家可以使用clone()方法,没有什么区别;

  2. 对拷贝过的数组进行排序;

  3. 使用HashMap存储排序后的各个元素的位置;

  4. 可以利用map.containsKey()方法来判断某个元素是否重复出现,如果重复出现,则跳过;

  5. 使用count来记录重复元素的个数;

    因为这里元素重复之后其下标值不会增加

    如:[1,3,4,5,5,6]对应的下标为[1,2,3,4,4,5],而不是[1,2,3,4,4,6]

  6. 遍历初始数组,将每个元素对应的排序后的下标存储到结果数组res[]中;

  7. 返回结果数组,完成;

AC 代码

class Solution {
    public int[] arrayRankTransform(int[] arr) {
        if(arr.length == 0) return new int[arr.length];
        int[] temp = new int[arr.length];       //1
        for(int i = 0;i < arr.length;i++){
            temp[i] = arr[i];
        }
        Arrays.sort(temp);                      //2
        int[] res = new int[arr.length];
        Map<Integer,Integer> map = new HashMap<>();
        int count = 0;
        for(int i = 0;i < temp.length;i++){     //3
            if(map.containsKey(temp[i])){       //4
                count++;                        //5
                continue;
            }
            map.put(temp[i],i+1-count);
        }
        int index = 0;
        for(int ele :arr){                      //6
            res[index++] = map.get(ele);
        }
        return res;                             //7
    }
}

参考

先排序,用对象obj存排序好的index,再遍历原始数组去找obj里面的value是多少并push到result里面 - 数组序号转换 - 力扣(LeetCode)