简单数位dp

题目链接

题意：[n,m]中不包含4和62的数的个数

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

int dp[10][10]; // dp[i][j]表示最高位数字为i，长度为j的的数字串中满足无4，无62的串的总数

void Init()
{
    memset(dp,0,sizeof(dp));
    //只有一位时
    for(int i = 0; i <= 9; i++)  if(i != 4) dp[i][1] = 1;
    //先枚举两位数字的，最后枚举7位数字的
    for(int j = 2; j <= 8; j++)
    {
        //第j位取值为i
        for(int i = 0; i <= 9; i++)
        {
            //第j-1位取值为k
            for(int k = 0; k <= 9; k++)
            {
                if(i != 4 && !(i == 6 && k == 2)) dp[i][j] += dp[k][j - 1];
            }
        }
    }
}

//求得小于n的串中满足不包括4且不包括62这个条件的串个总数
int calu(int n)
{
    int digit[10],cur = 1,x = n;
    while(x)
    {
        digit[cur] = x%10;
        x = x/10;
        cur++;
    }
    digit[cur] = 0;
    //从最高位枚举到最低位
    int ans = 0;
    for(int i = cur - 1; i >= 1; i--)
    {
        //枚举第i位
        for(int k = 0; k < digit[i]; k++)
        {
            if(k != 4 && !(digit[i+1] == 6 && k == 2))
                ans += dp[k][i];
        }
        if(digit[i] == 4 || (digit[i] == 2 && digit[i + 1] == 6)) break;
    }
    return ans;
}
int main()
{
    Init();
    int n,m;
    while(scanf("%d%d" ,&n,&m))
    {
        if(n == 0 && m == 0) break;
        int x = calu(n);
        int y = calu(m + 1);
        printf("%d\n",y - x);
    }
    return 0;
}