hdu3555 Bomb 数位DP入门

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
using namespace std;
long long int dp[30][3];
void init()
{
dp[0][0]=1;dp[0][1]=0;dp[0][2]=0;
for(int i=1;i<30;i++)
{
dp[i][0]=dp[i-1][0]*10-dp[i-1][1];//dp[i][0]表示长度为i的不含49的数的个数
dp[i][1]=dp[i-1][0];//dp[i][1]表示长度为i的不含49且最高位为9的数的个数
dp[i][2]=dp[i-1][2]*10+dp[i-1][1];// dp[i][2]表示长度为i的含49 的数的个数
}
}
int bit[30];
long long int solve(long long int n)
{
int len=1;
while(n)
{
bit[len++]=n%10;
n/=10;
}
bit[len]=0;
long long int ans=0;
bool flag=0;
for(int i=len;i>=1;i--)
{
ans+=dp[i-1][2]*bit[i];
if(flag) ans+=dp[i-1][0]*bit[i];//如果高位已经出现49那么后面随意
if(flag==0 && bit[i]>4) ans+=dp[i-1][1];
if(bit[i+1]==4 && bit[i]== 9) flag=1;

}
if(flag ) ans++;
return ans;
}
int main()
{
int t;
long long int n;
scanf("%d",&t);
init();
while(t--)
{
scanf("%I64d",&n);

cout<<solve(n)<<endl;
}
return 0;
}