leetcode 386. 字典序排数

给定一个整数 n, 返回从 1 到 n 的字典顺序。

例如,

给定 n =1 3,返回 [1,10,11,12,13,2,3,4,5,6,7,8,9] 。

请尽可能的优化算法的时间复杂度和空间复杂度。 输入的数据 n 小于等于 5,000,000。

这道题直观的想法是,建一个vector,和一个vector,int转换到string,排序后再转回int,空间复杂度为O(N),时间复杂度为O(N*lgN)。但是还要加上to_string和stoi的开销,还是蛮大的。

略加思索,可以用DFS,优先扩展深度,即可得到1,10,100,101,11,110…………的排序。
代码如下:

class Solution {
public:
    vector<int> lexicalOrder(int n) {
        vector<int>work;
        for(int i=1;i<=9;++i){
            if(i<=n){
                work.push_back(i);
                helper(work,n,i*10);
            }else{
                break;
            }
        }
        return work;
    }
    void helper(vector<int>&work,int n,int sum){
        for(int i=0;i<=9;++i){
            int tmp=sum+i;
            if(tmp<=n){
                work.push_back(tmp);
                helper(work,n,tmp*10);
            }else{
                break;
            }
        }
        return;
    }
};

你可能感兴趣的:(leetcode,c++)