leetcode算法学习(17)—— 将整数按权重排序

题目描述

我们将整数 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 的权重为 9（12 --> 6 --> 3 --> 10 --> 5 --> 16 --> 8 --> 4 --> 2 --> 1）
13 的权重为 9
14 的权重为 17
15 的权重为 17
区间内的数按权重排序以后的结果为 [12,13,14,15] 。对于 k = 2 ，答案是第二个整数也就是 13 。
注意，12 和 13 有相同的权重，所以我们按照它们本身升序排序。14 和 15 同理。

示例 2：
输入：lo = 1, hi = 1, k = 1
输出：1

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

示例 4：
输入：lo = 10, hi = 20, k = 5
输出：13

示例 5：
输入：lo = 1, hi = 1000, k = 777
输出：570

方法一：递归

思路

记 x 的权重为 f(x)，按照题意很明显我们可以构造这样的递归式：

于是我们就可以递归求解每个数字的权重了。

代码

class Solution:
    def getKth(self, lo: int, hi: int, k: int) -> int:
        def get_weight(n):
            if n==1:return 0
            if n%2==0:return get_weight(n/2)+1
            else: return get_weight(3*n+1)+1
        return sorted(list(range(lo,hi+1)),key=lambda x:(get_weight(x),x))[k-1]

方法二：字典

思路

• 通过字典存放整数及其权重
  • 若整数最初就为1，权重为0
  • 若 x 是偶数，那么 x = x / 2，若x 是奇数，那么 x = 3 * x + 1，每个过程权重均增加1
• 排序：先按照值排序在按照键排序

代码

class Solution:
    def getKth(self, lo: int, hi: int, k: int) -> int:        
        dic={}
        for item in range(lo,hi+1):
            i=0
            old_item=item
            if item==1: i=0
            while item!=1: 
                i+=1
                item=(item/2) if item%2==0 else (3*item+1)
            dic[old_item]=i
        return sorted(dic.items(),key=lambda x:(x[1],x[0]))[k-1][0]

方法三：记忆化(字典+递归)

思路

我们知道在求 f(3)的时候会调用到 f(10)，在求f(20) 的时候也会调用到 f(10)。同样的，如果单纯递归计算权重的话，会存在很多重复计算，我们可以用记忆化的方式来加速这个过程，即==「先查表，再计算」和「先记忆，再返回」。我们可以用一个哈希映射==作为这里的记忆化的「表」，这样保证每个元素的权值只被计算 1次。

代码

class Solution:
    def getKth(self, lo: int, hi: int, k: int) -> int:
        dic={1:0}
        def get_weight(n):
            if n in dic:
                return dic[n]
            dic[n]=(get_weight(n/2)+1) if n%2==0 else (get_weight(3*n+1)+1)
            return dic[n]
        return sorted(list(range(lo,hi+1)),key=lambda x:(get_weight(x),x))[k-1]