字典序排数
- 题目描述
- 递归法
- 迭代
题目描述
难度 - 中等
leetcode386. 字典序排数
给你一个整数 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 * 104
递归法
首先容易想到使用「递归」来实现 DFS。
将[1,n]的数按照字典序添加到答案,本质上是对一颗节点数量为 ,形态类似字典树的多阶树进行遍历,根节点为0,需要被跳过,因此我们可以从树的第二层开始搜索。
树中每个节点的值为其搜索路径所代表的数字,且每个节点有[0,9]共10 个子节点。
代码演示:
List<Integer> ans = new ArrayList<>();
int _n;
public List<Integer> lexicalOrder(int n) {
_n = n;
for (int i = 1; i <= 9; i++) dfs(i);
return ans;
}
void dfs(int cur) {
if (cur > _n) return ;
ans.add(cur);
for (int i = 0; i <= 9; i++) dfs(cur * 10 + i);
}
迭代
共有 个数需要被处理,假设当前处理到的数为j ,根据字典序规则,在满足条件的前提下,我们优先在j 的后面添加 0(即j * 10 < n 满足),否则我们考虑将上一位回退并进行加一操作。
代码演示:
public List<Integer> lexicalOrder(int n) {
List<Integer> ans = new ArrayList<>();
for (int i = 0, j = 1; i < n; i++) {
ans.add(j);
if (j * 10 <= n) {
j *= 10;
} else {
while (j % 10 == 9 || j + 1 > n) j /= 10;
j++;
}
}
return ans;
}