给你一个整数 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;
}
}