[思维]数字1到n的字典序第k小

Aurora141592

于 2021-01-08 13:55:36 发布

阅读量227

点赞数

分类专栏：思维

本文链接：https://blog.youkuaiyun.com/wulalalawu/article/details/112358380

版权

思维专栏收录该内容

14 篇文章

订阅专栏

这篇博客介绍了一个高效的算法模板，用于在接近O(logn)的时间复杂度内找到1到n之间字典序排序的第k小数字。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

存个很强的模板，貌似已经快到(O(logn))了

ll n, k, cnt = 1;
        scanf("%lld %lld", &n, &k);
        k--;
        while(k){
            ll st = 0, head = cnt, tail = cnt + 1;
            while(head <= n){
                st += min(n + 1, tail) - head;
                head *= 10;
                tail *= 10;
            }
            if(st <= k){
                ++cnt;
                k -= st;
            }else{
                cnt *= 10;
                --k;
            }
        }
        printf("%lld\n",cnt);