数组序号转换

116 阅读1分钟

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

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 <= 10^5
  • -10^9 <= arr[i] <= 10^9

解题

/**
 * @param {number[]} arr
 * @return {number[]}
 */
var arrayRankTransform = function (arr) {
  let nums = Array.from(new Set(arr));
  nums.sort((a, b) => a - b);
  let res = new Array(arr.length);
  const findIndexOf = (n) => {
    let left = 0;
    let right = nums.length - 1;
    while (left <= right) {
      let mid = (left + right) >> 2;
      if (nums[mid] === n) {
        return mid;
      } else if (nums[mid] > n) {
        right = mid - 1;
      } else {
        left = mid + 1;
      }
    }
  };
  for (let i = 0; i < arr.length; i++) {
    res[i] = findIndexOf(arr[i]) + 1;
  }
  return res;
};