hdu3555 数位dp

题意：你求出[1，n]中包含49的数的个数；

数位dp

1.dp[i][0] 代表数字长度为len不含49的个数 

2.dp[i][1] 代表数字长度为len不含49但是以9开头的个数

3.dp[i][2] 代表数字长度为len含有49的个数

代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#include <stdlib.h>
#include <string.h>
#include <iomanip>
#define N 500010
#define INF 10000000
#define LL long long
#define eps 10E-9
#define mem(a)  memset(a,0,sizeof(a))
#define w(a)   while(a)
#define s(a)   scanf("%d",&a)
#define ss(a,b)   scanf("%d%d",&a,&b)
#define sss(a,b,c)   scanf("%lld%lld%lld",&a,&b,&c)
#define MAXN 9999
#define MAXSIZE 10
#define DLEN 4
#define MAXN 9999
#define MAXSIZE 10
#define DLEN 4
using namespace std;
LL dp[22][3] = {0};
LL digit[22];
void init(){
    dp[0][0] = 1;
    for(int i=1; i<21; i++){
        dp[i][0] = dp[i-1][0]*10 - dp[i-1][1];// 在dp[i-1][0]前面添上0~9,但是要减去dp[i-1][1]    因为4和9构成49
        dp[i][1] = dp[i-1][0];//直接在dp[i-1][0]前面添上9就行了
        dp[i][2] = dp[i-1][2]*10 + dp[i-1][1];//在dp[i-1][2]前面添上0~9,或者在dp[i-1][1]前面添上4
    }
}
LL solve(LL x){
     mem(digit);
     LL len = 0;
     while(x){
        digit[++len] = x%10;
        x/=10;
     }
     digit[len+1]=0;
     LL ans=0;
     bool flag = false;
     for(int i=len; i; i--){
          ans += dp[i-1][2]*digit[i];
          if(flag)
            ans+=dp[i-1][0]*digit[i];
          if(!flag && digit[i]>4)
            ans+=dp[i-1][1];
          if(digit[i+1] == 4 && digit[i] == 9)
            flag = true;
     }
     return  ans;
}
int main(){
    int  t;
    LL n;
    init();
    w(~s(t)){
        w(t--){
           cin>>n;
           cout<<solve(n+1)<<endl;
        }
    }
    return 0;
}