蓝桥杯每日一题----第k个数

题目:第k个数

主要为了学习数字的构造方法,如何快速求1~n中前缀为pre的数字的个数。

题目分析

一开始想的是把数字转化为字符串,然后丢给sort排序就行了,但是n太大了,会出现溢出问题。走到这里也从侧面反映了对所有的数字排序然后输出第k个数字的方法是不可取的,那么可不可以直接构造出第k个数字呢?

首先查找一下以1开头的数字个数,假设是在1~321里面寻找,那么以1开头的数字有1+10+100=111个,如果k=13<111,说明可以确定排序为k的数字的第一位就是1,那么再枚举下一位,下一位枚举以10开头的数字个数(这里1是已经确定的,但是呢1也是以1开头的,在枚举10的时候相当于舍弃了数字1,所以这里要k--,减掉数字1)有1+10=11个,假设k=12>11,那么说明排序为k的数字的第二位不是0,那么按照字典序枚举下一个数字,但是此时,我们可以把以10开头的数字舍弃掉了,所以k-=11,k=1。即查找以11开头的数字的个数有1+10=11个,假设k<11,说明排名为k的数字是以11开头的,然后再找110开头的数字的个数是1个,刚好和k相等,说明我已经找到那个数字了,就是110。

我们自己按照字典序排一下,看一看是不是,

1,10,101,102,103,104,105,106,107,108,109,11,110

可以看到110刚好排在第13的位置,说明我们上面分析的过程是对的。接下来看一下刚刚那个代码应该怎么写。

java 复制代码
 Scanner scanner = new Scanner(System.in);
 int n = scanner.nextInt();//数据范围为1~n
 int k = scanner.nextInt();//找字典序排名为第k的数字
  int cur = 1;//从以数字1开头的数字开始尝试
  while(k > 0) {
        long nums = get(n,cur,cur+1);//找以cur开头的数字的个数,具体怎么找后续讨论
        if(k==nums&&k==1) {//如果k和nums相等或者不相等,但是k=1了,都说明当前的数字cur就是我们要找的数字,
            System.out.println(cur);
            return;
        }
        if(nums < k) {//当前数字个数小于k,说明不是以cur开头的,就找下一个,即cur+1,同时排除掉以cur开头的数字,所以k-=num
            cur += 1;
            k -= nums;
        }else {//当前数字个数大于k,说明是以cur开头的,那么就令cur*=10,然后再接着找下一位应该是以啥开头,cur*=10说明我们舍弃了cur本身,所以这里k要减1
            cur *= 10;
            k -= 1;
        }
    }

接下来就是如何求以cur开头的数字个数了

方法一

假设求1~123中以cur=1开头的数字个数,这里再加一个辅助的cur1,cur1=cur+1=2,

数字位数为1位时:cur1-cur=2-1=1;

cur1*=10;cur*=10;

数字位数为2位时:cur1-cur=20-10=10;

cur1*=10;cur*=10;

数字位数为3位时:cur1-cur=200-100=100;(×)

我们达不到200,只能达到123,所以应该是

数字位数为3位时:min(cur1,n+1)-cur=123+1-100=24;

总结:

数字位数为p位时:num+=min(cur1,n+1)-cur;

所以这里的代码为,

java 复制代码
private static long get(int n, long cur, long cur1) {
    // TODO Auto-generated method stub
    long nums = 0;//统计以cur开头的数字的个数
    while(cur <= n) {
        nums += Math.min(n+1, cur1)-cur;
        cur *= 10;
        cur1 *= 10;
    }
    return nums;
}

方法二

假设求1~123中以cur=1开头的数字个数,

base=cur=1;max=cur=1;

数字位数为1位时:max-base+1=1-1+1=1;

base*=10=10;max=max10+9=19;
数字位数为2位时:max-base+1=19-10+1=10;
base
=10=100;max=max*10+9=199;

数字位数为3位时:max-base=199-100+1=100;(×)

我们达不到200,只能达到123,所以应该是

数字位数为3位时:min(max,n)-base+1=123-100+1=24;

总结:

数字位数为p位时:num+=min(max,n)-base+1;

所以这里的代码为,

java 复制代码
private static long get2(int n, long cur) {//13 1
    // TODO Auto-generated method stub
    long nums = 0;//统计以cur开头的数字的个数
    long base = cur;//19 13-10+1
    long max = cur;//10
    while(base <= n) {
        nums += Math.min(n, max)-base+1;
        max =max * 10+9;//和base位数相同但是是该位数下以cur开头的最大的数字
        base *= 10;
    }
    return nums;
}

全部代码

java 复制代码
import java.util.Scanner;

public class Main{
public static void main(String[] args) {
    Scanner scanner = new Scanner(System.in);
    int n = scanner.nextInt();
    int k = scanner.nextInt();
    int cur = 1;
//    k--;
    while(k > 0) {
//        long nums = get(n,cur,cur+1);
        long nums = get2(n,cur);
        if(k==nums&&k==1) {
            System.out.println(cur);
            return;
        }
//        System.out.println(cur + " " + nums);
        if(nums < k) {//当前字符个数小于k
            cur += 1;
            k -= nums;
        }else {
            cur *= 10;
            k -= 1;
        }
    }
    System.out.println(cur);
}

private static long get(int n, long cur, long cur1) {
    // TODO Auto-generated method stub
    long nums = 0;//统计以cur开头的数字的个数
    while(cur <= n) {
        nums += Math.min(n+1, cur1)-cur;
        cur *= 10;
        cur1 *= 10;
    }
    return nums;
}

private static long get2(int n, long cur) {//13 1
    // TODO Auto-generated method stub
    long nums = 0;//统计以cur开头的数字的个数
    long base = cur;//19 13-10+1
    long max = cur;//10
    while(base <= n) {
        nums += Math.min(n, max)-base+1;
        max =max * 10+9;//和base位数相同但是是该位数下以cur开头的最大的数字
        base *= 10;
    }
    return nums;
}
}

测试结果

相关推荐
算法歌者20 分钟前
[算法]入门1.矩阵转置
算法
林开落L34 分钟前
前缀和算法习题篇(上)
c++·算法·leetcode
远望清一色35 分钟前
基于MATLAB边缘检测博文
开发语言·算法·matlab
tyler_download37 分钟前
手撸 chatgpt 大模型:简述 LLM 的架构,算法和训练流程
算法·chatgpt
SoraLuna1 小时前
「Mac玩转仓颉内测版7」入门篇7 - Cangjie控制结构(下)
算法·macos·动态规划·cangjie
我狠狠地刷刷刷刷刷1 小时前
中文分词模拟器
开发语言·python·算法
鸽鸽程序猿1 小时前
【算法】【优选算法】前缀和(上)
java·算法·前缀和
九圣残炎1 小时前
【从零开始的LeetCode-算法】2559. 统计范围内的元音字符串数
java·算法·leetcode
YSRM1 小时前
Experimental Analysis of Dedicated GPU in Virtual Framework using vGPU 论文分析
算法·gpu算力·vgpu·pci直通
韭菜盖饭2 小时前
LeetCode每日一题3261---统计满足 K 约束的子字符串数量 II
数据结构·算法·leetcode