刷题-力扣-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;
    }

};
posted @ 2022-04-18 19:41  韩亚光  阅读(36)  评论(0编辑  收藏  举报