hdu 3709（数位dp 记忆化搜索）

平衡数

题目链接：点击打开链接

// 这道题比较难吧 和以往的数位记忆化搜索设计的状态不太一样
// 也是用的记忆化dfs 熟练之后应该就没太大问题了

#include<stdio.h>
#include<string.h>

__int64 dp[20][20][2010];// pos 数位位置  v支点的位置 s 力矩的数值
int digit[20];

__int64 dfs(int pos,int v,int s,int doing )// 因为当前位置前一位是多少对这一个状态没有影响 只有力矩会影响到当前状态 所以去掉pre 加上了s
{
    if(pos==-1)
        {
            return s==0?1:0;//如果枚举到最后力矩为0 是平衡数
        }
    if(!doing && dp[pos][v][s]!=-1)return dp[pos][v][s];
    if(s<0)return dp[pos][v][s]=0;// 当前力矩小于0的话以后力矩都会小于0

    int i,End;
    __int64 ans=0;
    End=doing?digit[pos]:9;

    for(i=0;i<=End;i++)
    {
        ans+=dfs(pos-1,v,s+(pos-v)*i,doing&&i==End);
    }
    if(!doing )
        dp[pos][v][s]=ans;
    return ans;
}
__int64 solve(__int64 x)
{
    if(x<0)return 0;
    int l;
    l=0;
    memset(dp,-1,sizeof(dp));
    memset(digit,0,sizeof(digit));

    while(x>0)
    {
      digit[l++]=x%10;
      x=x/10;
    }

    int i;
    __int64 ans=0;
    for(i=l-1;i>=0;i--)
    {
        ans+=dfs(l-1,i,0,1);
    }
    return ans-(l-1);//长度为l的数000…00会被重复计算l次(每个支点一次) 需减去（l-1）次
}
int main()
{
    int t;
    __int64 a,b;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%I64d %I64d",&a,&b);
        printf("%I64d\n",solve(b)-solve(a-1));
    }
    return 0;
}