hdu 2089 数位dp入门题

#include<stdio.h>
//dp[i][0]代表不存在不吉利数字
//dp[i][1]代表不存在不吉利数字但是以2开头
//dp[i][2]代表存在不吉利数字
#define N 10
int dp[N][3];
void init() {
int i;
dp[0][0]=1;
for(i=1;i<10;i++) {
    dp[i][0]=dp[i-1][0]*9-dp[i-1][1];//当前不存在不吉利数字=上一个的不存在不吉利数字*非四-可能有不吉利数字以二开头
    dp[i][1]=dp[i-1][0];//上一个长度不吉利数字个数添加个2
    dp[i][2]=dp[i-1][2]*10+dp[i-1][0]+dp[i-1][1];//上一个状态的不吉利数字前面添加任意数+上一个状态的不存在不吉利数字前面加上4+上一个状态的不存在不吉利数字但是以二开头前面加上六
}
return ;
}
int slove(int x) {
       int a[N],i,j,k,flag=0,to=x;
       k=0;
       while(x) {
        a[++k]=x%10;//储存起来
        x/=10;
       }
       a[k+1]=0;
       j=0;//保存不吉利数字的个数
       for(i=k;i>0;i--) {
        j=j+dp[i-1][2]*a[i];//a[i]个i-1不吉利数字
        if(flag)j=j+dp[i-1][0]*a[i];//如果前面存在不吉利数字那么就随意了还要再加上当前的吉利数字的个数
        else {
            if(a[i]>4)j=j+dp[i-1][0];//如果存在a[i]等于四的情况，那么就要加i-1长度的吉利数字的个数
            if(a[i]>6)j=j+dp[i-1][1];//当存在a[i]大于六的情况，还要加上长度为i-1的以二开头的吉利数字的个数
             if(a[i+1]==6&&a[i]>2)j=j+dp[i][1];//如果当前可能为2的话，要加上当以二开头时的吉利数字的个数
        }
        if((a[i+1]==6&&a[i]==2)||a[i]==4)//标记最高位已出现过不吉利数字
            flag=1;
       }
       return to-j;
}
int main() {
    int n,m;
    init();
    while(scanf("%d%d",&n,&m),n||m) {
        printf("%d\n",slove(m+1)-slove(n));
    }
return 0;
}