hdoj 3709 Beautiful numbers

题目链接：Balanced Number

题目大意：给你n和m，问你这个范围里面的平衡数有多少，平衡数是指以某一个数位位对称轴，两边距离乘数位算贡献，如果相等，则位平衡数

题目思路：枚举对称轴，数位dp，算距离不取绝对值，小于零剪枝，需要注意的是0,00,000这样的数都不是平衡数，但是数位dp的时候我们枚举进去了，所以需要减去

#include <map>#include <set>#include <queue>#include <stack>#include <cmath>#include <vector>#include <cstdio>#include <cstring>#include <cstdlib>#include <iostream>#include <algorithm>using namespace std;typedef long long ll;ll digit[20],dp[20][5000][20];ll dfs(ll pos,ll num,ll pivot,bool limit){    if(pos == -1) return num == 0?1:0;    if(num < 0) return 0;    if(!limit&&dp[pos][num][pivot] != -1) return dp[pos][num][pivot];    ll up = limit?digit[pos]:9;    ll ans = 0;    for(ll i = 0;i <= up;i++){        ans += dfs(pos-1,num+i*(pos-pivot),pivot,limit&&i == digit[pos]);    }    if(!limit) dp[pos][num][pivot] = ans;    return ans;}ll solve(ll x){    ll pos = 0;    ll res = 0;    if(x == -1) return 0;    while(x){        digit[pos++] = x%10;        x/=10;    }    for(ll i = 0;i < pos;i++)        res += dfs(pos-1,0,i,true);    return res-pos+1;}int main(){    ll n,m,t;    memset(dp,-1,sizeof(dp));    scanf("%lld",&t);    while(t--){        scanf("%lld%lld",&n,&m);        printf("%lld\n",solve(m)-solve(n-1));    }    return 0;}