HDU 3555 Bomb(数位DP)

HDU 3555 Bomb(数位DP)
http://acm.hdu.edu.cn/showproblem.php?pid=3555
题意:给你一个N,现在要你求区间[1,N]内包含49序列的数的个数.
分析:简单数位DP.
令f[i][j]=x表示长i的且以j为最高位的数中不包含49的数有多少个.

注意:本题数据是long long,各种数据类型都要变.
如果用f[i][j]表示长i最高位为j的包含49的数有多少个的话,那么
f[i+1][4] += (10^(i-1) - f[i][9]) 这样的话就不太好计算了.
AC代码:

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
long long f[50][10];//f[i][j]表示长为i且最高位为j的数中不含49的数有多少个
int digit[50];
void init()
{
    memset(f,0,sizeof(f));
    f[0][0]=1;
    for(int i=1;i<=49;i++)
        for(int j=0;j<=9;j++)
            for(int k=0;k<=9;k++)if(!(j==4&&k==9))
            {
                f[i][j] += f[i-1][k];
            }
}
long long cal(long long n)//[0,n]中不含49的数有多少个.
{
    int len=0;
    while(n)
    {
        digit[++len]=n%10;
        n/=10;
    }
    digit[len+1]=0;
    if(len==0) return 1;

    int i;
    long long ans=0;
    for(i=len;i>=1;i--)
    {
        for(int j=0;j<digit[i];j++)if( !(digit[i+1]==4&&j==9) )//这句别忘了
            ans += f[i][j];
        if(digit[i+1]==4 && digit[i]==9)//已经有49了,直接退出
            break;
    }
    if(i==0) ans++;//本身这个数字也合法
    return ans;
}
int main()
{
    int T;
    long long n;
    scanf("%d",&T);
    init();
    while(T--)
    {
        scanf("%I64d",&n);
        printf("%I64d\n",n+1-cal(n));
    }
    return 0;
}