难度中等288
给你一个整数 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 * 10^4
思路:假设我们现在有一个按序生成的满足条件的数number,生成下一个数的时候,为了使得字典序最小,因此我们会尝试在后面加一个0,也就是number*10。
A.那么如果number*10<=n,很显然生成的下一个数就应该是number*10。
B.如果number*10>n,此时我们想要字典序保持最小,那么应该动的是低位的数,即尝试number+1,此时会遇到一些特殊情况,如
当我们number=19,n=23时,我们发现number+1=20,然而2的字典序比19大比20小,因此,当number % 10 == 9 时,不仅仅会引起高位进位,还会在后面带一个0,因此我们需要把这个0去掉,也就是number=(number + 1) / 10。
不仅仅如此,考虑边界条件,如果number = 16,n=16,当前number+1=17>n,因此低位已经到了最大,此时应该动比当前位更高的一位。
现在我们整理一下算法:设当前满足要求的数为number
1.初始化的时候,number=1,循环n次
2.若number*10<=n,则number=number*10,再进循环;否则转3
3.若number%10==9或number+1>n,此时应该明确需要动的是当前位的上一位,因此number/=10(注意这边需要用循环来做,因为可能遇到2999999,此时它的下一个生成数应该是3)
class Solution {
public:
vector<int> lexicalOrder(int n) {
vector<int> nums_vec(n);
int num = 1;
for(int i = 0; i < n; ++ i){
nums_vec[i] = num;
if(num * 10 <= n) {//后面接0得到的字典序毫无疑问是最小的
num *= 10;
} else {//如果后面不能接0
//情况1 接0后比n大,尝试不接0,尝试加1,可能有两种情乱,一种是最低位是9,会导致高位改变,并且后面多一个0,但是加1也可能比n大,此时只能改高位的字典序
while(num % 10 == 9 || num + 1 > n){
num /= 10;
}
num += 1;
}
}
return nums_vec;
}
};