[swustoj 1097] 2014

2014(1097)

问题描述

今年是2014年，所以小明喜欢2014的每一位数字（即：2,0,1,4）,小明想知道在区间[l，r](包括l和r)中有多少个数中含有这4个数字（数字无前缀零）。

输入

多组数据。

每组数据输入2个数l，r（0<l<r<=10^9）

输出

输出占一行，即区间[l，r]（包括l和r）中包含的满足条件的数的个数

样例输入

1 10
100 1024

样例输出

0

1

简单数位DP

#include <iostream>

#include <cstdio>

#include <cstring>

#include <cmath>

using namespace std;



int bit[20];

int dp[20][2][2][2][2];



int dfs(int pos,int n2,int n0,int n1,int n4,bool limit,bool fzero)

{

    if(pos==-1)

    {

        return n2&n0&n1&n4;

    }

    if(!limit && !fzero && dp[pos][n2][n0][n1][n4]!=-1) return dp[pos][n2][n0][n1][n4];

    int end=limit?bit[pos]:9;

    int ans=0;

    for(int i=0;i<=end;i++)

    {

        int nn2=i==2?1:0;

        int nn0=i==0?1:0;

        int nn1=i==1?1:0;

        int nn4=i==4?1:0;

        if(fzero) nn0=0;

        ans+=dfs(pos-1,n2|nn2,n0|nn0,n1|nn1,n4|nn4,limit && i==end,fzero && !i);

    }

    if(!limit && !fzero) dp[pos][n2][n0][n1][n4]=ans;

    return ans;

}

int cal(int n)

{

    int len=0;

    while(n)

    {

        bit[len++]=n%10;

        n/=10;

    }

    return dfs(len-1,0,0,0,0,1,1);

}

int main()

{

    int l,r;

    memset(dp,-1,sizeof(dp));

    while(cin>>l>>r)

    {

        cout<<cal(r)-cal(l-1)<<"\n";

    }

    return 0;

}