日新刷题 - 1331. 数组序号转换

141 阅读1分钟

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

一 描述

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

给你一个整数数组 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 <= 10^5
  • -10^9 <= arr[i] <= 10^9

二 分析

我一开始的思路是:先对数组排序,然后找到每个元素对应的rank,之后再生成答案ans。

上面的思路需要Map来做映射,但是要是可以用数组来实现就更好了。

先找到最大的元素和最小的元素,那么用来对应rank的数组大小就是max - min + 1 将每个元素对应到rank数组,将arr存在的元素位置变为1 最后将arr中的每个元素对应到其的rank值。

三 答案

//本质先计算每个元素对应的rank位置

class Solution {
    public int[] arrayRankTransform(int[] arr){

        int max = Integer.MIN_VALUE, min = Integer.MAX_VALUE, j = 1;

        for(int i : arr){
            if(i > max) max = i;
            if(i < min) min = i;
        }

        int[] bucket = new int[max - min + 1];

        for(int i : arr){
            bucket[i - min] = 1;
        }

        for(int i = 0; i < bucket.length; i++){
            if(bucket[i] != 0){
                bucket[i] = j++;
            }
        }

        int[] ans = new int[arr.length];

        int index = 0;
        for(int i:arr){
            ans[index++] = bucket[i - min];
        }
        return ans;
    }
}

参考

2行代码,思路简单,性能高效接近100 - 数组序号转换 - 力扣(LeetCode) (leetcode-cn.com)