$Poj3208$ 启示录 数位统计$DP$

Poj  AcWing

Description

Sol 

这题长得就比较像数位$DP$叭.

所以先用$DP$进行预处理,再基于拼凑思想,通过"试填法"求出最终的答案.

设$F[i][3]$表示由$i$位数字构成的魔鬼数有多少个,$F[i][j](0<=j<=2)$表示由$i$位数字组成的,开头有$j$个$6$的非魔鬼数有多少个.注意,在计算$F[i][j]$时允许前导$0$的存在

$F[i][0]=9*(F[i-1][0]+F[i-1][1]+F[i-1][2])$

$F[i][1]=F[i-1][0]$

$F[i][2]=F[i-1][1]$

$F[i][3]=F[i-1][2]+10*F[i-1][3]$

预处理完成之后,我们先通过$F[i][3]$确定第$X$小魔鬼数的位数,然后从左到右进行试填,试填的数从小到大枚举

Code

#include<iostream>
#include<cstdio>
#include<cmath>
#define Rg register
#define il inline
#define db double
#define ll long long
#define go(i,a,b) for(Rg int i=a;i<=b;i++)
#define yes(i,a,b) for(Rg int i=a;i>=b;i--)
using namespace std;
il int read()
{
    int x=,y=;char c=getchar();
    while(c<''||c>''){if(c=='-')y=-;c=getchar();}
    while(c>=''&&c<=''){x=(x<<)+(x<<)+c-'';c=getchar();}
    return x*y;
}
int T,n,d,k;
ll f[][];
il void init()
{
    f[][]=;
    go(i,,)
    {
        f[i][]=*(f[i-][]+f[i-][]+f[i-][]);
        f[i][]=f[i-][];
        f[i][]=f[i-][];
        f[i][]=f[i-][]+*f[i-][];
    }
}
int main()
{
    init();T=read();
    while(T--)
    {
        n=read();d=;k=;
        while(f[d][]<n)d++;
        yes(i,d,)
        {
            go(j,,)
            {
                //求若第i位为j,那么剩下i-1位有多少种填法能使这个数成为魔鬼数
                ll ct=f[i-][];
                if(j== || k>=)
                    go(t,max(,-k-(j==)),)ct+=f[i-][t];
                if(ct<n)n-=ct;//应该填一个更大的数
                else //就填j
                {
                    if(j==)k++;else if(k<)k=;
                    printf("%d",j);break;
                }
            }
        }
        printf("\n");
    }
    return ;
}