洛谷 P2022 有趣的数

题目描述

让我们来考虑1到N的正整数集合。让我们把集合中的元素按照字典序排列，例如当N=11时，其顺序应该为：1,10,11,2,3,4,5,6,7,8,9。

定义K在N个数中的位置为Q(N,K)，例如Q(11,2)=4。现在给出整数K和M，要求找到最小的N，使得Q(N,K)=M。

输入输出格式

输入格式：

输入文件只有一行，是两个整数K和M。

输出格式：

输出文件只有一行，是最小的N，如果不存在这样的N就输出0。

输入输出样例

输入样例#1：
Sample 1: 2 4
Sample 2: 100000001 1000000000
这里Sample 1 和 2是分开的两个数据点。
输出样例#1：
Sample 1: 11
Sample 2: 100000000888888879
说明

【数据约定】

40%的数据，1<=K,M<=10^5；

100%的数据，1<=K,M<=10^9。

分析

对于比N小的，字典序N的数ans=k/10（位数的次方）-1 (可以自己列几个数出来看看就好了）

代码

#include <iostream>
#include <cstdio>

#define N 100

using namespace std;

long long s[N], f[N];
long long n, m, k;
long long cnt = 0;
long long Base = 1;

void get(long long x) {
    int t = 0;

    while(x) {
        s[++t] = x % 10;
        x /= 10;
        Base *= 10;
    }

    Base /= 10;

    for(int i = 1; i <= t; i++)
        f[i] = s[t - i + 1];

    cnt += t - 1;

    for(int i = t; i >= 1; i--) {
        long long now = 0;

        for(int j = 1; j <= i; j++) {
            if(j != 1)
                now = now * 10 + f[j];
            else now = now * 10 + f[j] - 1;
        }

        cnt += now;
    }
}

int main() {
    std::cin >> k >> m;

    get(k);

    if(cnt >= m || k == Base && cnt < m - 1) {
        std::cout << 0 << std::endl;
        return 0;
    }

    long long x = k - Base, y = k;

    for(; cnt < m - 1;) {
        x *= 10;
        y *= 10;
        cnt += x;
    }

    long long ans = max(k, y - (cnt - m + 2));
    std::cout << ans << endl;
}