清北学堂模拟赛d6t3 反击数

分析：显然是一道数位dp题，不过需要一些奇怪的姿势.常规的数位dp能统计出一个区间内满足条件的数的个数，可是我们要求第k个，怎么办呢？转化为经典的二分问题，我们二分当前数的大小，看它是第几大的，就可以了.

显然数位dp套上模板，再用上kmp的next数组就可以了，传递4个参数：还剩下多少位没有匹配，匹配了多少位，是否达到上限和是否匹配成功，到最后判断一下就可以了.

学到了一种很强的思想：如果能求出i是第几个数，要求出第k个数就可以二分i的值.

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

long long L, R, K, f[][][];
int m, nextt[], a[];
char X[];

void init()
{
    nextt[] = ;
    nextt[] = ;
    for (int i = ; i < m; i++)
    {
        int j = nextt[i];
        while (j && X[i] != X[j])
            j = nextt[j];
        nextt[i + ] = X[i] == X[j] ? j +  : ;
    }
}

long long dfs(int len, int w, bool limit, bool flag)
{
    if (len == )
        return flag;
    if (!limit && f[len][w][flag] != -)
        return f[len][w][flag];
    int maxn = limit ? a[len] : ;
    long long cnt = ;
    for (int i = ; i <= maxn; i++)
    {
        int t = w;
        while (t && X[t] - '' != i)
            t = nextt[t];
        if (X[t] - '' == i)
            t++;
        cnt += dfs(len - , t, limit && (i == a[len]), flag || (t == m));
    }
    return limit ? cnt : f[len][w][flag] = cnt;
}

long long query(long long u)
{
    int cnt = ;
    while (u)
    {
        a[++cnt] = u % ;
        u /= ;
    }
    memset(f, -, sizeof(f));
    return dfs(cnt, , , );
}

int main()
{
    scanf("%lld %lld %s %lld", &L, &R, X, &K);
    m = strlen(X);
    init();
    if (query(R) < K + query(L - ))
    {
        printf("Hey,wake up!\n");
        return ;
    }
    long long t = query(L - ),ans = L;
    long long l = L, r = R;
    while (l < r)
    {
        long long mid = (l + r) >> ;
        if (query(mid) - t >= K)
        {
            r = mid;
            ans = mid;
        }
        else
            l = mid + ;
    }
    printf("%lld\n", r);

    return ;
}