HDU 3555 Bomb

RT.

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
long long t,n,bit[],ret=,dp[][][];
void get_bit(long long x)
{
    ret=;
    while (x) {bit[++ret]=x%;x/=;}
}
long long dfs(long long pos,long long pre,bool flag1,long long flag2)
{
    if (!pos) return flag1;
    if ((!flag2) && (~dp[pos][pre][flag1])) return dp[pos][pre][flag1];
    long long ans=,up=flag2?bit[pos]:;
    for (long long i=;i<=up;i++)
        ans+=dfs(pos-,i,flag1|((pre==) && (i==)),flag2&&i==up);
    if (!flag2) dp[pos][pre][flag1]=ans;
    return ans;
}
int main()
{
    memset(dp,-,sizeof(dp));
    scanf("%lld",&t);
    for (long long i=;i<=t;i++)
    {
        scanf("%lld",&n);
        get_bit(n);printf("%lld\n",dfs(ret,,,));
    }
    return ;
}