hdu 3709 数位dp

数位dp，有了进一步的了解，模板也可以优化一下了

题意：找出区间内平衡数的个数，所谓的平衡数，就是以这个数字的某一位为支点，另外两边的数字大小乘以力矩之和相等，即为平衡数
例如4139，以3为支点
4*2 + 1*1 = 9 and 9*1 = 9，称为平衡数

Sample Input
2
0 9
7604 24324

Sample Output
10
897

 

#include<cstdio>
#include<cstring>
using namespace std;
__int64 dp[19][19][2000];
int digit[19];
__int64 dfs(int p,int o,int s,bool e) {       //数位，支点位置，前面到支点的和，是否任意填
    if (p==-1) return s==0;
    if(s<0)    return 0;
    if (!e &&dp[p][o][s]!=-1) return dp[p][o][s];
    __int64 res = 0;
    int u = e?digit[p]:9;
    for (int d=0;d<=u;++d)
    {
        int ns=d*(p-o)+s;
        res+=dfs(p-1,o,ns,e&&d==u);
    }
    return e?res:dp[p][o][s]=res;
}
__int64 solve(__int64 n)
{
    int len=0;
    while(n)
    {
        digit[len++]=n%10;
        n/=10;
    }
    __int64 ans=0;
    for(int i=0;i<len;i++)
    {
        ans+=dfs(len-1,i,0,1);
    }
    return ans-len+1;
}
int main()
{
    int t;
    __int64 n,m;
    //freopen("1.in","r",stdin);
    memset(dp,-1,sizeof(dp));
    scanf("%d",&t);
    while(t--)
    {
        scanf("%I64d%I64d",&n,&m);
        printf("%I64d\n",solve(m)-solve(n-1));
    }
}