386. Lexicographical Numbers

问题描述

Given an integer n, return 1 - n in lexicographical order.

For example, given 13, return: [1,10,11,12,13,2,3,4,5,6,7,8,9].

Please optimize your algorithm to use less time and space. The input size may be as large as 5,000,000.

题目链接：

---

思路分析

给一个数字n，将1-n的数字按照字典顺序输出。

字典顺序就是小的在前，大的在后。首先生成一个含有n个元素的res数组。然后开始从1开始存入数据。首先是向1后面乘10，如果乘不动了，再开始加1；如果加超过了n，说明数太大了，除10之后继续，类似于dfs的回溯。同时对于加1后的元素有一个判断，去掉尾部的0们，因为这时的cur是通过+1产生的，太大，需要去掉0，他们应该通过cur = cur * 10 来产生。

代码

class Solution {
public:
    vector<int> lexicalOrder(int n) {
        vector<int> res(n);
        int cur = 1;
        for(int i = 0; i < n; i++){
            res[i] = cur;
            if(cur * 10 <= n)
                cur *= 10;
            else{
                if(cur >= n)
                    cur /= 10;
                cur += 1;
                while (cur % 10 == 0)
                    cur /= 10;
            }
        }
        return res;
    }
};

时间复杂度：$O(n)$
空间复杂度：$O(n)$

---

反思

一开始对于字典顺序排序有理解上的偏差。第一次做出来的不对。

同样，这样带有回溯性质的问题，可以使用dfs来做。从1-9分别递归遍历，用合适的值继续递归。

class Solution {
public:
    vector<int> lexicalOrder(int n) {
        vector<int> res;
        for(int i = 1; i < 10; i++){
            dfs(i, n, res);
        }
        return res;
    }

    void dfs(int cur, int n, vector<int>& res){
        if (cur > n)
            return;
        res.push_back(cur);
        for(int i = 0; i < 10; i++){
            if(cur * 10 + i > n)
                return;
            dfs(cur * 10 + i, n, res);
        }
    }
};