Leetcode 386. 字典序排数

给你一个整数 n ，按字典序返回范围 [1, n] 内所有整数。

你必须设计一个时间复杂度为 O(n) 且使用 O(1) 额外空间的算法。

题解

题解转载自斌HZB
方法一：深度优先搜索

如果我们将 h 位的所有整数看做如下所示 h 层的 k 叉树（k≤10），[1,n] 范围内的所有整数的字典序实际上就是这棵 k 叉树的先序遍历顺序。

因此，我们可以从 1 开始进行深度优先搜索（注意第一位不能为0），在搜索过程中不断将当前整数添加至答案中，再从 2 开始搜索，以此类推。如果搜索过程中当前数大于 n，直接返回。

class Solution {
    //十叉树的递归遍历
    public List<Integer> lexicalOrder(int n) {
        List<Integer> list = new ArrayList<>();
        //记住不需要添加0
        for (int i = 1; i < 10; i++){
             dfs(n, i, list);
        }
        return list;
    }

    private void dfs(int n,int i,List<Integer> list){
        //递归终止条件
        if(i>n) return;
        list.add(i);
        for(int j=0;j<=9;j++){
            dfs(n,i*10+j,list);
        }
    }
}

方法二：迭代

通常而言，将DFS改为迭代需要额外的栈空间。注意到当前搜索的数 num 本身就能反映出每一层递归的状态，例如当前搜索至 123，我们知道当前层遍历至 3，上一层遍历至 2，再上一层遍历至 1。如果我们需要返回上一层，只需要令 num//=10 即可。

依照此规律，我们可以不需要额外栈空间，直接将DFS改写为迭代：

class Solution {
    public List<Integer> lexicalOrder(int n) {

        List<Integer> res = new ArrayList<>();
        int num = 1;
        //第一个循环必须是判断res.size() < n而不是判断num<=n
        //e.g. num=14>13,则会因为14>13直接跳出循环，不能退回到2
        while(res.size() < n){
            while(num<=n){
                res.add(num);
                //*10相当于 遍历子节点
                num *= 10;
            }
            while(num % 10 == 9 || num > n){
                //如果当前层子节点 遍历完了，或者 是不存在的节点(因为已经大于n了)，则返回上一层(/=10)
                num /= 10;
            }
            //当前层的下一个节点
            num += 1;
        }
        return res;
    }
}