刷题力扣386. 字典序排数

题目链接

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/lexicographical-numbers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

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

题目分析

1. 根据题目描述，按照字母序排列[1,n]
2. 深度优先搜索，假设cur表示当前数字，若cur10不大于n，令cur=cur10，否则cur+1
3. 边界条件，若cur+1大于n，则令cur=cur/10；若cur的个位数为9，即cur+1是10的倍数，令cur=cur/10。执行该步骤直到cur的个位数不为9。
4. 例如，n为3000，cur为2889（下一位289、2890），cur自增变为2890，而2890应该出现在289的后面。

代码

class Solution {
public:
    vector<int> lexicalOrder(int n) {
        std::vector<int> nums;
        int cur = 1;
        for (int i = 0; i < n; ++i) {
            nums.emplace_back(cur);
            if (cur * 10 > n) {
                // 若cur为2889，cur自增变为2890，而2890应该出现在289的后面，所以2889/10
                while (cur + 1 > n || cur % 10 == 9) {
                    cur /= 10;
                }
                ++cur;
            } else {
                cur *= 10;
            }
        }
        return nums;
    }

};