「每日LeetCode」2021年7月17日

本文最后更新于:2023年3月19日 晚上

Lt1331. 数组序号转换

1331. 数组序号转换

给你一个整数数组 arr ,请你将数组中的每个元素替换为它们排序后的序号。
序号代表了一个元素有多大。序号编号的规则如下:

  • 序号从 1 开始编号。
  • 一个元素越大,那么序号越大。如果两个元素相等,那么它们的序号相同。
  • 每个数字的序号都应该尽可能地小。

示例 1:

1
2
3
输入:arr = [40,10,20,30]
输出:[4,1,2,3]
解释:40 是最大的元素。 10 是最小的元素。 20 是第二小的数字。 30 是第三小的数字。

示例 2:

1
2
3
输入:arr = [100,100,100]
输出:[1,1,1]
解释:所有元素有相同的序号。

示例 3:

1
2
输入:arr = [37,12,28,9,100,56,80,5,12]
输出:[5,3,4,2,8,6,7,1,3]

提示:

  • 0 <= arr.length <= 10
  • -10 <= arr[i] <= 10

思路

升序排序后,用哈希表记录所有元素对应的序号,再将原数组转化为对应的序号即可。

解答

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
/**
* @param {number[]} arr
* @return {number[]}
*/
var arrayRankTransform = function (arr) {
const sortArr = arr.slice();
sortArr.sort((a, b) => a - b);
const map = new Map();
let count = 1;
for (let i = 0; i < sortArr.length; i++) {
const element = sortArr[i];
if (!map.has(element)) map.set(element, count++);
}
return arr.map((e) => map.get(e));
};