开启我的LeetCode刷题日记:440. 字典序的第K小数字

118 阅读3分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

编程世界总是离不了算法

最近在看框架源码时,会有很多算法的实现逻辑,有时候会感到吃力

于是决定蹭着假期,加强算法和数据结构相关的知识

那怎么提升呢?

其实我知道算法这东西没有捷径,多写多练才能提升,于是我开启我的LeetCode刷题之旅

第一阶段目标是:200道,每天12

为了不乱,本系列文章目录分为三部分:

  1. 今日题目:xxx
  2. 我的思路
  3. 代码实现

今天题目:440. 字典序的第K小数字

给定整数 n 和 k,返回  [1, n] 中字典序第 k 小的数字。

 

示例 1:

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

输入: n = 1, k = 1 输出: 1  

提示:

1 <= k <= n <= 109

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/k-… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

我的思路

有选择地访问字典树:向右访问兄弟结点或者向下访问第一个子结点。 令初始数字cur为1,初始位序order为1,做如下操作:

获取以当前数字为前缀的数字个数num; 判断当前位序order加上num的个数是否大于k, 如果是的话则说明第k小的数字在cur的子结点,令cur成为它的第一个子结点cur=cur*10,位序加一order=order+1; 否则说明第k小的数字在cur的兄弟结点,令cur成为它的下一个兄弟结点cur=cur+1,位序加上子结点数量order=order+num; 若order == k 则说明,当前的数字cur就是我们要找的数字;否则重复步骤1,2。

代码实现

var findKthNumber = function(n, k) {
    let cur = 1, order = 1;//cur:当前数字, order:当前数字的位序
    while(order !== k){//当order与k相等时,说明cur就是我们想要的答案
        let num = getNumByPrefix(cur,n);//求以cur为前缀的数字个数
        if(k < order + num){
            //向下查找
            cur *= 10;//cur更新为它的第一个子结点
            order ++;           
        }
        else{
            //向右查找
            cur ++;//cur更新为它右边第一个兄弟兄弟结点
            order += num;
        }
    }
    return cur;
    //求以x为前缀的数字(包含x)得个数,范围为[1,limit]
    function getNumByPrefix(x,limit){
        let count = 0;//数字个数
        let left = x, right = x;//左值和右值默认初始化为前缀x
        while(left<=limit){
            count += Math.min(right, limit) - left +1;//计数加上每一层的结点数量
            left *=10, right = right*10 + 9;//左值left和右值right的更新式
        }
        return count;
    }
};



总结

实现方式其实有很多,这里仅供参考~

由于刚开始刷题,也不知道从哪里刷好,如果前辈们有好的建议,希望不吝赐教,感谢🌹