python leetcode 440. K-th Smallest in Lexicographical Order

本文介绍了一种解决LexicographicalNumbers问题的高效算法,通过构建10叉树并采用类似先序遍历的方式,实现对大规模数字序列的快速定位,避免了传统遍历方法在大数据量下可能出现的超时问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

如果按照Lexicographical Numbers这道题进行遍历取值显然会超时(n最大可能为10^9 也就是说必须要用30位bit存 30*10^9 数据量是非常大的)
先来看看n=100 1,10,100,11,12,13,14,15,16,17,18,19,2… 1下面对应10,11,…,19, 10下面对100这样就对应于一个10叉树 类似先序遍历顺序即为序号
cur先为1 要快速定位到第K小所在的位置
譬如1到2中间隔了多少数字step=1+10+1

  • first=1 last=2
  • step+=1
  • first=10 last=20
  • step+=10
  • first=100 last=200
  • step=101-100 因为last>100
    然后判断k是否在1到2的步数中
  • 如果在 则cur=10*cur k-=1
  • 如果不在 cur=cur+1 k-=step

在这里插入图片描述

class Solution(object):
    def findKthNumber(self, n, k):
        """
        :type n: int
        :type k: int
        :rtype: int
        """
        cur=1
        k-=1
        while k>0:
            step,first,last=0,cur,cur+1
            while first<=n:
                step+=min(last,n+1)-first
                first*=10
                last*=10
            if k>=step:
                k-=step
                cur+=1
            else:
                cur*=10
                k-=1 
        return cur
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值