题目描述

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

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

示例 1:

输入:n = 13
输出:[1,10,11,12,13,2,3,4,5,6,7,8,9]

示例 2:

输入:n = 2
输出:[1,2]

提示:

  • 1 <= n <= 5 * 10^4

这道题就是相当于英语词典中的按首字母排序的单词顺序。所以我们来看一下官方的解法。

官方

class Solution {
    public List<Integer> lexicalOrder(int n) {
        List<Integer> ans = new ArrayList<>();
        //从1开始加入
        int number = 1;
        for(int i=0;i<n;i++){
            ans.add(number);
            //尝试添加0
            if(number*10<=n)number*=10;
            else{
            //判断是否到当前序列的末尾
                while(number%10==9||number+1>n)number/=10;
                number++;
            }
        }
        return ans;
    }
}

有大佬直接造出一棵字典树,虽然使用了额外空间,但确实是一个好思路。

class Solution {
    List<Integer> ans = new ArrayList<>();
    public List<Integer> lexicalOrder(int n) {
       for(int i=1;i<=9;i++) dfs(i,n);
       return ans;
    }
    //构建一棵字典树,进行深度优先搜索
    public void dfs(int cur,int limit){
        //大于n则直接返回空值
        if(cur>limit) return ;
        ans.add(cur);
        //另开辟10个dfs分支
        for(int i=0;i<=9;i++) dfs(cur*10+i,limit);
    }
}