【题解】hdu2089 数位DP

题目链接
dp[i][j]表示开头是j的i位数中的吉利数个数
dp[i][j]=dp[i-1][k]其中j!=4且当j=6时k!=2
从高到低枚举哪一位比n的哪一位小

#include<cstdio>
#include<cstring>
int dp[10][10],n,m;
int bit[10];//存每一位的值 
void init()
{
    dp[0][0]=1;
    for(int i=1;i<=7;i++)
    for(int j=0;j<=9;j++)
    {
        if(j==4)continue;
        for(int k=0;k<=9;k++)
        {
            if((j==6&&k==2)||k==4)continue;
            dp[i][j]+=dp[i-1][k];
        }
    }
}
int solve(int len)
{
    int ans=0;
    bit[len+1]=0;
    for(int i=len;i>0;i--)
    {
        for(int j=0;j<bit[i];j++)
        {
            if(j==2&&bit[i+1]==6)continue;
            ans+=dp[i][j];
        }
        if(bit[i]==4||(bit[i]==2&&bit[i+1]==6))break;
    }
    return ans;
}
int main()
{
    //freopen("in.txt","r",stdin);
    init();
    while(~scanf("%d%d",&n,&m)&&n+m)
    {
        m++;//因为只算了[1,r)
        int t1=n,t2=m,l1=0,l2=0;
        while(t1)
        {
            l1++;
            bit[l1]=t1%10;//把数压入数组中 
            t1/=10;
        } 
        t1=solve(l1);
        while(t2)
        {
            l2++;
            bit[l2]=t2%10;
            t2/=10;
        }
        t2=solve(l2);
        printf("%d\n",t2-t1);
    }
    return 0;
}