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;

}