数组序号转换

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第 34 天，点击查看活动详情

问题描述

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

思路分析

首先我们要先理解一下题目的意思，题目会给我们一个数组，我们计算数组中每个元素排序后的序号。

• 1、获取排序后的数组

我们需要使用一个新变量来保存排序后的数组，使用数组解构及sort方法可以快速完成。

let newArr = [...arr].sort((a,b)=>{
    return a - b;
});

• 2、使用哈希表记录每个元素的序号

使用哈希表来记录每个元素的序号，序号的计算需要满足以下规则：

• 序号从 1 开始编号。

• 一个元素越大，那么序号越大。如果两个元素相等，那么它们的序号相同。

• 每个数字的序号都应该尽可能地小

let map = {};
let ind = 1;
for(let i = 0; i < newArr.length; i++){
    if(!map[newArr[i]]) map[newArr[i]] = ind++;
}

• 3、将数组中的元素转换为序号

前面我们已经将每个元素的下标计算出来并保存到了哈希表中，这里我们只需要通过哈希表将元素对应的序号获取即可。

for(let i = 0; i < arr.length; i++){
    arr[i] = map[arr[i]];
}

AC 代码

完整代码如下：

/**
 * @param {number[]} arr
 * @return {number[]}
 */
var arrayRankTransform = function(arr) {
    let newArr = [...arr].sort((a,b)=>{
        return a - b;
    });
    let map = {};
    let ind = 1;
    for(let i = 0; i < newArr.length; i++){
        if(!map[newArr[i]]) map[newArr[i]] = ind++;
    }
    for(let i = 0; i < arr.length; i++){
        arr[i] = map[arr[i]];
    }
    return arr;
};

说在后面

本人为算法业余爱好者，平时只是随着兴趣偶尔刷刷题，如果上面分享有错误的地方，欢迎指出，感激不尽。