题目描述:给定整数 n
和 k
,返回 [1, n]
中字典序第 k
小的数字。
示例 1:
输入: n = 13, k = 2
输出: 10
解释: 字典序的排列是 [1, 10, 11, 12, 13, 2, 3, 4, 5, 6, 7, 8, 9],所以第二小的数字是 10。
示例 2:
输入: n = 1, k = 1
输出: 1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/k-th-smallest-in-lexicographical-order
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
-------------------------------------------------------------------------------------------------------------------------
day4:今日题比较困难,思路挺好想到的,就是实现挺难的,不容易想到,本题为字节跳动常考Top10之一,难度挺高的,继续加油吧
代码如下:
public int findKthNumber(int n, int k) {
int curr = 1;
k--;
while (k > 0) {
int steps = getSteps(curr, n);
if (steps <= k) {
k -= steps;
curr++;
} else {
curr = curr * 10;
k--;
}
}
return curr;
}
public int getSteps(int curr, long n) {
int steps = 0;
long first = curr;
long last = curr;
while (first <= n) {
steps += Math.min(last, n) - first + 1;
first = first * 10;
last = last * 10 + 9;
}
return steps;
}
解题思路:
设置curr为返回值,即k下标对应的结果,默认值为1,对应只输入1,开局k--是减掉1,因为1是绝对存在的,当只有1时就while循环不进入,直接返回1 ;
当k值大于1时,进入循环,通过getSteps获取当前数对应每一组数的个数,参数为curr和n,curr表示当前结果所在位数,n表示需要的位数,通过循环就能获得curr所在层每层的个数,如n为365,curr为1,第一次执行的返回结果为111,表示在以1开头,到365结尾的情况下,1开头所占的个数有111个,即1+10+100,同理2也是111,当curr判断为3时,因为有Math.min(last,n)的存在,表明当前层已经占不满了,只能占余数个,即1+10+65+1个(含3、30-39、300-365,300也是一个,所以要加1);
求出curr首位后再通过k不断的调用getSteps方法,来获取后面的几位(如上面例子所提到的十位、个位),最终加起来就是答案
再提供一种思路:我们知道n的位数x,通过除以和取余x个1的操作,可以更快速的算出区间,明天写出来了就更新。
错误考虑:通过列表来装下所有字符,通过ascii码比较来排序,最后通过get返回对应值,结果代码如下:
public static int findKthNumber(int n, int k) {
List<String> list = new ArrayList<>(); //4289384 1922239
for (int i = 1; i <= n; i++) {
list.add(""+i);
}
Collections.sort(list);
return Integer.valueOf(list.get(k-1));
}
简洁度高,但是超出内存限制了,数太多了不行