LeetCode Everyday - 将整数按权重排序

74 阅读2分钟

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

将整数按权重排序

我们将整数 x 的 权重 定义为按照下述规则将 x 变成 1 所需要的步数:

如果 x 是偶数,那么 x = x / 2 如果 x 是奇数,那么 x = 3 * x + 1 比方说,x=3 的权重为 7 。因为 3 需要 7 步变成 1 (3 --> 10 --> 5 --> 16 --> 8 --> 4 --> 2 --> 1)。

给你三个整数 lo, hi 和 k 。你的任务是将区间 [lo, hi] 之间的整数按照它们的权重 升序排序 ,如果大于等于 2 个整数有 相同 的权重,那么按照数字自身的数值 升序排序 。

请你返回区间 [lo, hi] 之间的整数按权重排序后的第 k 个数。

注意,题目保证对于任意整数 x (lo <= x <= hi) ,它变成 1 所需要的步数是一个 32 位有符号整数。

示例1:

输入:lo = 12, hi = 15, k = 2
输出:13
解释:12 的权重为 912 --> 6 --> 3 --> 10 --> 5 --> 16 --> 8 --> 4 --> 2 --> 113 的权重为 9
14 的权重为 17
15 的权重为 17
区间内的数按权重排序以后的结果为 [12,13,14,15] 。对于 k = 2 ,答案是第二个整数也就是 13 。
注意,1213 有相同的权重,所以我们按照它们本身升序排序。1415 同理。

示例2:

输入:lo = 7, hi = 11, k = 4
输出:7
解释:区间内整数 [7, 8, 9, 10, 11] 对应的权重为 [16, 3, 19, 6, 14] 。
按权重排序后得到的结果为 [8, 10, 11, 7, 9] 。
排序后数组中第 4 个数字为 7

提示:

  • 1 <= lo <= hi <= 1000
  • 1 <= k <= hi - lo + 1

解题思路:

1.  先遍历lo到hi之间的数,使用递归求出权
2.  求出的权放到map 中
3.  对map进行排序
4.  通过k 找到要找的元素

我的答案:

/**
 * @param {number} lo
 * @param {number} hi
 * @param {number} k
 * @return {number}
 */
var getKth = function(lo, hi, k) {
    let quan = 0;
    let map = new Map();
    let fun = function(num) {
      if(num === 1) {
        map.set(i, quan);
        quan = 0;
        return;
      }
      if(map.get(num)) {
        quan += map.get(num);
        map.set(i, quan);
        quan = 0;
        return;
      } else {
        if(num % 2 === 0) {
          num = num / 2;
        } else {
          num = 3 * num + 1;
        }
        quan ++;
        fun(num);
      }

    }
    for(var i = lo ; i <= hi; i ++) {
      fun(i);
    }
    let arr = [...map];
    arr.sort((a,b) => a[1] - b[1]);
    return arr[k - 1][0];
};

最后

如果有更好的解法或者思路, 欢迎在评论区和我交流~ ღ( ´・ᴗ・` )