hdu 3555 Bomb（数位dp）

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

__int64 dp[25][3];

void Init()
{
memset(dp,0,sizeof(dp));
int i;
dp[0][0]=1;
for(i=1;i<=20;i++)
{
dp[i][0]=dp[i-1][0]*10-dp[i-1][1];
dp[i][1]=dp[i-1][0];
dp[i][2]=dp[i-1][2]*10+dp[i-1][1];
}
}

__int64 solve(__int64 n)
{
__int64 i,len=0,ans,flag,digit[25];
memset(digit,0,sizeof(digit));
while(n)
{
digit[++len]=n%10;
n/=10;
}
ans=0;
flag=0;
for(i=len;i>0;i--)
{
//printf("%d ",ans);
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=1;
}
}
return ans;
}

int main()
{
Init();
int T;
__int64 n;
scanf("%d",&T);
while(T--)
{
scanf("%I64d",&n);
printf("%I64d\n",solve(n+1));
}
return 0;
}