hdu 3555数位dp基础入门题

#include<stdio.h>
#define N 20
long long  dp[N][3];
void init(){
long long  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];
}
}
long long  slove(long long  x) {
long long  i,a[N],k,flag=0;
long long j;
k=0;
while(x) {
    a[++k]=x%10;
    x/=10;
}
a[k+1]=0;
j=0;
for(i=k;i>=1;i--) {
    j=j+a[i]*dp[i-1][2];
    if(flag)j=j+a[i]*dp[i-1][0];
    else
        if(a[i]>4)j=j+dp[i-1][1];
    if(a[i+1]==4&&a[i]==9) flag=1;
}
return j;
}
int main() {
  long long n,t;
  init();
  scanf("%I64d",&t);
  while(t--) {
    scanf("%I64d",&n);
    printf("%I64d\n",slove(n+1));//必须加一，因为如果是n的话他只能枚举到n-1
  }
return 0;
}