【HDOJ 2089】不要62

【HDOJ 2089】不要62

第一个数位dp的题 做的老困难了。。。不过好歹是做出来了 迈出了第一步。。

对大牛来说这种题都是小case

代码如下:

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

int dp[8][3];
/*
dp[i][0]无不吉利数字
dp[i][1]无不吉利数字且高位为2
dp[i][2】有不吉利数字
*/

void Init()
{
    memset(dp,0,sizeof(dp));
    dp[0][0] = 1;
    int i;
    for(i = 1; i <= 8; ++i)
    {
        dp[i][0]=dp[i-1][0]*9-dp[i-1][1];  //在最高位加上除了4之外的9个数字，但是可能在2之前加了6  
        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，或者在2前面加4
    }
}

int Solve(int n)
{
    int ls[9],len = 0,i,ans,tmp = n,flag = false;
    while(n)
    {
        ls[++len] = n%10;
        n /= 10;
    }
    ans = ls[len+1] = 0;
    for(i = len; i; --i)
    {
        ans += dp[i-1][2]*ls[i];
        if(flag) ans += dp[i-1][0]*ls[i];//高位已出现4或62 后面随意加
        if(!flag && ls[i] > 4) ans += dp[i-1][0];//高位有出现死的可能
        if(!flag && ls[i+1] == 6 && ls[i] > 2) ans += dp[i][1];//高位有出现62的可能
        if(!flag && ls[i] > 6) ans += dp[i-1][1];
        if(ls[i] == 4 || (ls[i+1] == 6 && ls[i] == 2)) flag = 1;//出现4或62
    }
    return tmp - ans;
}

int main()
{
    int n,m;
    Init();
    while(~scanf("%d %d",&n,&m) && n && m)
    {
        printf("%d\n",Solve(m+1)-Solve(n));//[m,n)区间
    }
    return 0;
}