HDU - 3555 数位dp

题意：t 组样例 ，每组一个 n ,判断n以为含有49的数有多少个

#include<stdio.h>
#include<algorithm>
#include<iostream>
using namespace std;
typedef long long LL;
LL dp[25][3];
void init()
{
    dp[0][0]=1;//长度为 i 但是不含49的数有多少个
    dp[0][1]=0;//长度为 i 且最高位只有9的数有多少个
    dp[0][2]=0;//长度为 i 且包含49的数有多少个
    for(int i=1;i<=21;i++)
    {
        dp[i][0]=dp[i-1][0]*10-dp[i-1][1];//减去最高位为9的情况  它可以加上4
        dp[i][1]=dp[i-1][0];//在不含49的情况下 在高位加9
        dp[i][2]=dp[i-1][1]+dp[i-1][2]*10;//最高位只有9的情况加上4   短一位含有49的情况*10
    }
    return;
}
void work(long long n)
{
    LL ans=0;
    int bit[25],cnt=0;
    while(n)
    {
        bit[++cnt]=n%10;
        n/=10;
    }
    bit[cnt+1]=0;
    bool flag=0;
    for(int i=cnt;i>=1;i--)
    {
        ans+=dp[i-1][2]*bit[i];//低位中含有49
        if(!flag)/*没有经历过49*/
        {
            if(bit[i]>4)
                ans+=dp[i-1][1];/*最高位可以是4*/
        }
        else
            ans+=dp[i-1][0]*bit[i];/*49出现过之后 后面的数字不用在意*/
        if(bit[i+1]==4&&bit[i]==9)
            flag=1;
    }
    if(flag)
        ans++;
    printf("%lld\n",ans);
}
int main()
{
    LL t,n;
    init();
    scanf("%lld",&t);
    while(t--)
    {
        scanf("%lld",&n);
        work(n);
    }
    return 0;
}