字典序的第K小数字

最新推荐文章于 2022-06-07 19:54:18 发布

强殖装甲凯普

最新推荐文章于 2022-06-07 19:54:18 发布

阅读量181

点赞数

分类专栏：算法题

本文链接：https://blog.csdn.net/qq_38163755/article/details/116431340

版权

算法题专栏收录该内容

14 篇文章 0 订阅

订阅专栏

这篇文章介绍了解决LeetCode 440问题的方法，通过构建10叉树模型，计算每个节点的子树节点数，实现查找1到n范围内字典序第k小的整数。核心算法涉及递归和计数，适合对动态规划和搜索算法感兴趣的读者。

摘要由CSDN通过智能技术生成

文章目录

问题
解决

问题

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

注意：1 ≤ k ≤ n ≤ 109。

LeetCode 440

解决

本身是10个十叉树问题
在这里插入图片描述
我们从上往下，从左到右，每次计算以该节点为根节点的子树的节点数，判断其与k的关系，比k小说明不在子树里，k减去子树节点数，向右，值+1；比k大说明在子树里，k-1，要向下，值就要*=10

class Solution(object):
    def findKthNumber(self, n, k):
        """
        :type n: int
        :type k: int
        :rtype: int
        """
        res=1
        k-=1
        while k>0:
            cnt,cur,next=0,res,res+1
            while cur<=n:
                cnt+=min(n+1,next)-cur
                cur*=10
                next*=10
            if cnt<=k:
                k-=cnt
                res+=1
            else:
                k-=1
                res*=10
        return res