LeetCode刷题实战386:字典序排数
Given an integer n, return all the numbers in the range [1, n] sorted in lexicographical order.
You must write an algorithm that runs in O(n) time and uses O(1) extra space.
示例
给定 n =1 3,返回 [1,10,11,12,13,2,3,4,5,6,7,8,9] 。
请尽可能的优化算法的时间复杂度和空间复杂度。输入的数据 n 小于等于 5,000,000。
解题
class Solution {
public:
vector<int> lexicalOrder(int n) {
vector<int> nums(n,0);
for(int i=0;i<n;++i){//存入范围内的所有数字
nums[i]=i+1;
}
//使用自定义函数进行排序
sort(nums.begin(),nums.end(),[](int& lhs,int& rhs){
//先将两个整数转成字符串,使用字符串进行比较
string str1=to_string(lhs);
string str2=to_string(rhs);
int pos=0;
while(pos<str1.size()||pos<str2.size()){
if(pos<str1.size()&&pos<str2.size()){//都在范围内,比较字符的大小
if(str1[pos]<str2[pos]){
return true;
}
else if(str1[pos]>str2[pos]){//不在范围内,比较字符串的长度
return false;
}
}
else if(str1.size()<str2.size()){
return true;
}
else{
return false;
}
++pos;
}
return true;
});
return nums;
}
};