leetcode 440. 字典序的第K小数字(字节高频面试) dfs

还是要面向样例编程。。。
网上有很多比较好的思路,这是比较复杂但也是可以自己想出来的思路

给定整数 n 和 k,找到 1 到 n 中字典序第 k 小的数字。

注意:1 ≤ k ≤ n ≤ 1 0 9 10^9 109

示例 :

输入:
n: 13 k: 2

输出:
10

解释:
字典序的排列是 [1, 10, 11, 12, 13, 2, 3, 4, 5, 6, 7, 8, 9],所以第二小的数字是 10。

思路:


# 建立一个十叉树,然后去寻找第k小字典序的数字
# 怎么找呢
# 肯定是从节点1处开始遍历,如果从上到下遍历一遍的话,肯定会超时,那么我们需要判断k在不在这个根节点下,省的做无用功
# 如何判断呢
# 需要查找每个节点下的个数,比如判断1节点下的个数是否大于等于k,
# 如果小于,那么就k-=(1节点下的个数),然后去找2节点,
# 同样的,如果1节点下的个数大于等于k,就去找子节点,也是同样的道理。
# 查找时肯定不能暴力递归查找,应该遵从规律性,从层数出发,
# 比如查找1时,上面的层数是满十叉树,但是下面的肯定会被n截断,
# 比如n=102,那么对应的是1(第一层),10,11,。。19(第二层)100,101,102(第三层)到1这里被截断了,
#对每个节点处理时,可以分为三种情况,
# 未被截断,在数字内部被截断(比如上个例子中就是在1包括10的内部节点被截断的),以及已经被前面的数字截断,那么层数直接-1即可。
#具体看代码

class Solution:

    def dfs(self,p: int, n: int, k: int, n_num: int) -> int:
        if k == 1:
            return p
        p_num = int(0)
        temp =int( p) #当前节点在第几层
        while temp > 0:
            temp =int(temp/ 10)
            p_num += 1
        c =int( n_num - p_num)  # 判断还有几层
        # 判断是否被截断
        temp = int((p+1) * pow(10,c)-1) # 以p为前缀的最大的数字
        su = int(1)  # 当前节点的子节点数(包括自身)
        d=0
        if temp > n and n>=p*pow(10,c):  # 在中间被截断
            d=1
            su += n - p * pow(10,c) + 1  # 最后一层的个数
        if n<p*pow(10,c):#已经被前面的数字截断
            d=1
            su+=0
            #c -= 1
        if d==1:#被截断层数减一
            c-=1
        now = int(1)
        for i in range(c):
            now *= 10
            su += now#加上每一层的子节点数
        if k > su:  # 往右走
            k -= su
            return self.dfs(p + 1, n, k, n_num)
        else:  # 往下走
            k -= 1
            return self.dfs(p * 10, n, k, n_num)

    def findKthNumber(self, n: int, k: int) -> int:
        n_num = int(0)  # 判断到此几层
        temp = int(n)
        while temp > 0:
            temp =int(temp/10)
            n_num += 1
        return self.dfs(1, n, k, n_num)  # 开始从1查找




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值