386. 字典序排数
思路:给定一个整数 n, 返回从 1 到 n 的字典顺序。例如,给定 n =1 3,返回 [1,10,11,12,13,2,3,4,5,6,7,8,9] 。
请尽可能的优化算法的时间复杂度和空间复杂度。 输入的数据 n 小于等于 5,000,000。
涉及到字符串的排序,那最无脑的当然是用TreeMap啦~但是要注意遍历map的时候一定要转型呀,遍历的方法是这样:
Iterator iter = map.entrySet().iterator();
while(iter.hasNext()){
Map.Entry entry = (Map.Entry)iter.next();
Object key = entry.getKey();
Object value = entry.getValue();
}
这个遍历方法需要记住!!!
完整代码如下:
public List<Integer> lexicalOrder(int n) {
Map<String,Integer> map = new TreeMap<>();
for(int i=1;i<=n;i++){
map.put(String.valueOf(i),i);
}
List<Integer> res = new ArrayList<>();
Iterator iter = map.entrySet().iterator();
while(iter.hasNext()){
res.add((int)((Map.Entry)iter.next()).getValue());
}
return res;
}
另外,我看题解里都是用的DFS…我开始完全没往这方面想( ఠൠఠ )ノ
用DFS也比较巧妙,我们把1到9的每个数进行DFS,先被遍历的自然就是最小的,然后一次让该数去向十位百位…按这个顺序,遍历下来的也是字典序~
public List<Integer> lexicalOrder(int n) {
ArrayList<Integer> res = new ArrayList<>();
for(int i=1;i<10;i++){
dfs(n,i,res);
}
return res;
}
private void dfs(int n,int target,List<Integer> list){
if(target>n) return;
list.add(target);
target*=10;
for(int j=0;j<10;j++){
dfs(n,target+j,list);
}
}
注意dfs方法中,遍历需要从0开始而不是1,因为只有个位数才不允许第一位为0!!