HDU - 3709 Balanced Number（数位dp，注意前导零）

题目：给定区间[a,b]，求区间内平衡数的个数。所谓平衡数即有一位做平衡点，左右两边数字的力矩相等。

思路：dp[pos][k][sum]代表进行到pos位，以k位为平衡点时，加和为sum时的答案。

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
int t,k,wei[22];
ll A,B,dp[22][22][3600];
ll dfs(int pos,int sum,int limit,int lead)
{
    if(pos<1) return (sum==0&&!lead)?1:0;//判断一下前导零，不然0会被多加进去
    if(!limit&&!lead&&dp[pos][k][sum]!=-1)
        return dp[pos][k][sum];
    int up=limit?wei[pos]:9;
    ll ans=0;
    for(int i=0;i<=up;i++)
    {
        ans+=dfs(pos-1,sum+(pos-k)*i,limit&&i==up,i==0&&lead);
    }
    if(!limit&&!lead) dp[pos][k][sum]=ans;
    return ans;
}

ll solve(ll x)
{
    if(x==0) return 1;
    if(x<0) return 0;
    int len=0;
    while(x)
    {
        wei[++len]=x%10;
        x/=10;
    }
    ll ans=0;
    for(k=len;k>0;k--)
    {
        ans+=dfs(len,0,1,1);
    }
    return ans+1;///加上0的时候的答案
}
int main()
{
    memset(dp,-1,sizeof dp);
    scanf("%d",&t);
    while(t--)
    {
       scanf("%lld%lld",&A,&B);
       ll ans=solve(B)-solve(A-1);
       printf("%lld\n",ans);
    }
    return 0;
}