欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

440. 字典序的第K小数字

程序员文章站 2022-06-28 21:47:44
...

leetcode刷题

给定整数 n 和 k,找到 1 到 n 中字典序第 k 小的数字。
注意:1 ≤ k ≤ n ≤ 109。
示例 :
输入:
n: 13 k: 2
输出:
10
解释:
字典序的排列是 [1, 10, 11, 12, 13, 2, 3, 4, 5, 6, 7, 8, 9],所以第二小的数字是 10。

这是一道字节常考的算法题
**解题思路:10叉树
记录当前结点的个数,如果当前结点的个数num>k的话,则在子节点中去找,子节点就是x10,然后k要减去当前节点的个数num;如果当前结点数num<k的话,说明还太小呢,所以就向右走,cur++,k–;

**

440. 字典序的第K小数字

class Solution {
    public int findKthNumber(int n, int k) {
        //字典序列   10叉树

        //思路:前序遍历;每次记录子节点的数量num,如果num>k了,说明在子节点中;否则的话就在右边找

        int cur = 1;
        //1 ≤ k ≤ n ≤ 109
        while(k>1){
            int num = getCount(n,cur,cur+1);
            if(num<k){//如果小于了,则前序遍历-往右走
                cur++;
                k-=num;
            }else{//如果大于了,则在子节点中-往下走,,下的话就是10叉树
                cur*=10;
                k-=1;
            }
        }
        return cur;
    }
    private int getCount(int n,long first,long last){//设置为num,是因为*10之后怕溢出
        int count = 0;
        while(first <=n){
            count+=Math.min(last,n+1)-first;
            first*=10;
            last*=10;
        }
        return count;
    }
}
相关标签: leetcode 二叉树