zl程序教程

您现在的位置是:首页 >  其它

当前栏目

字典序第k小数字

数字 字典
2023-09-14 08:56:53 时间

 

 

 

 

https://leetcode.cn/problems/k-th-smallest-in-lexicographical-order/solution/pythonjavajavascriptgo-di-gui-by-himymbe-5mq5/

 

func findKthNumber(n int, k int) int {
    cur := 1
    k--
    for k > 0 {
        cnts := dfs(cur, cur,n)
        fmt.Println(cnts)
        if cnts <= k { //当前层cur,第cnts小的数比第k小还小,那么当前层的所有cnts个数都可以取,递归到下一层去找
            k -= cnts
            cur++
        } else {// 答案在当前节点的子节点中,则从当前cur层的最小子节点开始找,cur=cur*10
            k--
            cur *= 10
        }
    }
    return cur
}

func dfs (l, r,n int) int{//l,r分别是最小和最大子节点,返回值是递归开始层~当前层所有的子节点数
    //两个递归终止条件
    if l > n {
        return 0
    }
    if r > n {
        r = n
    }

    //当前层有 r - l + 1 个节点可取,递归到下一层。
    // l * 10: 从10变成100, r * 10 + 9: 从19变成199
    return r - l + 1 + dfs(l * 10, r * 10 + 9,n)
}