好数 数位dp

好数

Description

今年是2019年，如果从一个数中删去所有的3,4,5,6,7,8后剩下的数按照与按顺序排列刚好等于2019，则称这个数为好数。

例子：342560189，72019，2019是好数，12019，20219，9102456不是好数

注意：如果删去所有的3,4,5,6,7,8后剩下的数还有前导零的话，我们也不能去除掉这些前导零，比如3002019,500023019，在删除后都不是好数

输入n，请你求出1～n之间有多少个好数。

Input

第一行一个整数TT，代表有多少个询问，

接下来TT行，每行一个整数n

(1≤T≤105) (0≤n≤1018)

Output

对于每个询问输出一个整数：1～n之间有多少个好数。

Sample Input 1 

3
20199
22222
231456

Sample Output 1

13
19
265

#include<bits/stdc++.h>
#define fuck(x) std::cout<<"["<<#x<<"]:"<<x<<endl;
using namespace std;
typedef long long ll;

int s[20]= {0};

ll dp[20][2][2][2][2][2];
///   pos 2  0  1  9 fs0

ll dfs(int pos,bool hav2,bool hav0,bool hav1,bool hav9,bool fs0,bool lim)
{
    if(pos<=0)
    {
        if(hav2==1&&hav0==1&&hav1==1&&hav9==1)
            return 1;
        return 0;
    }
    if(!lim&&dp[pos][hav2][hav0][hav1][hav9][fs0]!=-1)
        return dp[pos][hav2][hav0][hav1][hav9][fs0];
    ll ans=0;
    int num= lim==0?9:s[pos];
    for(int i=0; i<=num; i++)
    {
        bool hav0s=hav0,hav1s=hav1,hav2s=hav2,hav9s=hav9,fs0s=fs0;

        if(fs0==1&&hav2==0&&(i==0||i==1||i==9))continue;
        if(hav0==0&&(i==1||i==9))continue;
        if(hav1==0&&(i==9))continue;

        if(hav2==1&&i==2)continue;
        if(hav0==1&&i==0)continue;
        if(hav1==1&&i==1)continue;
        if(hav9==1&&i==9)continue;

        if(hav2==0&&i==2)hav2s=1;
        if(hav0==0&&i==0&&fs0==1)hav0s=1;
        if(hav1==0&&i==1)hav1s=1;
        if(hav9==0&&i==9)hav9s=1;

        if(fs0==0&&i!=0)fs0s=1;

        ans+=dfs(pos-1,hav2s,hav0s,hav1s,hav9s,fs0s,lim&&i==num);
    }
    if(!lim)
        dp[pos][hav2][hav0][hav1][hav9][fs0]=ans;
    return ans;
}


ll solve(ll a)
{
    memset(s,0,sizeof(s));
    int i=0;
    while(a)
    {
        s[++i]=a%10;
        a/=10;
    }
    return dfs(i,0,0,0,0,0,1);
}

int main()
{
    int _;
    scanf("%d",&_);
    memset(dp,-1,sizeof(dp));
    solve((ll)1e18);
    while(_--)
    {
        ll a;
        scanf("%lld",&a);
        printf("%lld\n",solve(a));
    }
    return 0;
}

拿队友代码改的,另一个写法: 

#include<bits/stdc++.h>
#define LL long long

using namespace std;

const int maxn = 25;
LL dp[maxn][3000][2];
int a[maxn];

LL dfs(int pos,int pre,int flag,bool limit) {
    if(pos == -1)
        return pre == 2019;
    if(!limit && dp[pos][pre][flag] != -1)
        return dp[pos][pre][flag];
    int up = limit ? a[pos] : 9;
    LL ans = 0;
    for(int i = 0; i <= up; ++ i) {
        int pres=pre,flags=flag;

        if(flag==1&&pre==0&&i==0)continue;

        if(pre==0&&(i==1||i==9))continue;
        if(pre==2&&(i==2||i==1||i==9))continue;
        if(pre==20&&(i==2||i==0||i==9))continue;
        if(pre==201&&(i==2||i==0||i==1))continue;
        if(pre==2019&&(i==2||i==0||i==1||i==9))continue;

        if(flag==0&&i!=0)flags=1;

        if(pre==0&&i==2)pres=pre*10+i;
        if(pre==2&&i==0)pres=pre*10+i;
        if(pre==20&&i==1)pres=pre*10+i;
        if(pre==201&&i==9)pres=pre*10+i;

        ans+=dfs(pos-1,pres,flags,limit&&i==up);
    }
    if(!limit)
        dp[pos][pre][flag] = ans;
    return ans;
}

LL solve(LL x) {
    int pos = 0;
    while(x) {
        a[pos ++] = x % 10;
        x /= 10;
    }
    LL ans = 0;
    ans = dfs(pos - 1,0,0,1);
    return ans;
}

int main() {
    int t;
    scanf("%d",&t);
    solve(1e18);
    memset(dp,-1,sizeof(dp));
    while(t --) {
        LL n;
        scanf("%lld",&n);
        printf("%lld\n",solve(n));
    }
    return 0;
}