HDU 3709 Balanced Number （数位DP）

题意:
原题链接
一个数，如果能找到一个支点，使得这个树左右两边平衡（按照力矩计算），问[l,r]有多少，范围1e18

劳资终于会写出个数位DP了喵呜呜呜！！！

f(i,j,k)表示i位数，支点j，两边和位k（左边力为正，右边为负）
枚举j，注意减去0的情况，因为0是唯一一个可以变支点依然成立的数

可以用宏定义平移下标，当然也可以利用技巧左正右负使得一旦变成负数就退出。

代码：

#include<cmath>
#include<queue>
#include<cstdio>
#include<cctype>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
#define f(i,j,k) f[(i)][(j)][(k)+1800]
const int maxn=25;
int T;
int z[maxn],sz;
LL l,r,f[maxn][maxn][3600+105];
LL dp(int i,int j,int k,bool lim)
{
    if(i<1)return k==0;
    if(!lim && f(i,j,k)!=-1)return f(i,j,k);
    LL ret=0;
    int up=lim?z[i]:9;
    for(int l=0;l<=up;l++)ret+=dp(i-1,j,k+(i-j)*l,lim && l==up);
    if(!lim)return f(i,j,k)=ret;
    return ret;
}
LL calc(LL x)
{
    sz=0;
    do{
        z[++sz]=x%10;
        x/=10;
    }while(x);
    LL ret=0;
    for(int i=1;i<=sz;i++)ret+=dp(sz,i,0,1);
    return ret-sz+1;
}
int main()
{
    //freopen("in.txt","r",stdin);
    memset(f,-1,sizeof(f));
    for(scanf("%d",&T);T--;)
    {
        scanf("%lld%lld",&l,&r),
        printf("%lld\n",calc(r)-calc(l-1));
    }
    return 0;
}