【数位dp】HDU - 3709 Balanced Number

Problem Description

找出闭合区间内平衡数的个数。
平衡数：以某一位数为支点，左右两边位数以此乘积距离的和 相等。
例如4139：如果以3为支点 4*2+1*1 = 9*1 所以 4139是平衡数

思路:

简化问题：如果支点的位置是固定不变的，让我们求平衡数个数，就很好求。
这时候支点你是不确定在那个位置的，所以你可以枚举支点 进行记忆化搜索。对于每个不同的支点0都算了一遍，所以结果得减去长度 + 1

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll a[25], dp[25][25][2000];
ll dfs(int pos, int x, int num, int limit)
{
    if(pos < 0) return num == 0;
    if(num < 0) return 0;
    if(!limit && dp[pos][x][num] != -1) return dp[pos][x][num];
    int up = limit ? a[pos]:9;
    ll res = 0;
    for(int i = 0; i <= up; i++)
    {
        ll tmp = (pos - x) * i;
        res += dfs(pos-1, x, num+tmp, limit && i == a[pos]);
    }
    if(!limit) return dp[pos][x][num] = res;
    return res;
}
ll solve(ll x)
{
    if(x < 0) return 0;
    int pos = 0;
    ll ans = 0;
    while(x)
    {
        a[pos++] = x%10;
        x = x/10;
    }
    for(int i = 0; i < pos; i++)
    {
        ans += dfs(pos-1, i, 0, 1);
    }
    return ans - pos + 1;
}
int main()
{
    int T;
    cin >> T;
    ll lt, rt;
    memset(dp, -1, sizeof(dp));
    while(T--)
    {
        cin >> lt >> rt;
        cout << solve(rt) - solve(lt-1) << endl;
    }
    return 0;
}