HDU 3709 数位DP

题意

如果把一个数拆分为，左半部分，右边部分和平衡点，左半部分力矩=右半部分力矩。则称这个数为平衡数。问某个区间内有多少个平衡数。

题解

数位DP的经典套路。首先拆分数位，然后根据记忆化搜索。这道题由于数最多只有19位，所以可以暴力枚举平衡点，于是DP数组涉及到三个状态，当前长度，平衡点位置，总力矩。总力矩<0的时候就再也无法达到力矩平衡，可以直接return 0。

代码

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<string>
#include<set>
#include<map>
#include<bitset>
#define UP(i,l,h) for(int i=l;i<h;i++)
#define DOWN(i,h,l) for(int i=h-1;i>=l;i--)
#define W(a) while(a)
#define MEM(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define LL long long
#define MAXN 100010
#define EPS 1e-10
#define int LL

using namespace std;

int dig[25];
int dp[25][25][2000];

int dfs(int len,int pos,int sum,bool up){
    if(len<=0){
        return sum==0?1:0;
    }
    if(sum<0)
        return 0;
    if(!up&&(dp[len][pos][sum]!=-1))
        return dp[len][pos][sum];
    int n=up?dig[len]:9;
    int ans=0;
    UP(i,0,n+1)
        ans+=dfs(len-1,pos,(len-pos)*i+sum,up&(i==n));
    if(!up)
        dp[len][pos][sum]=ans;
    return ans;
}

int solve(int x){
    int len=0;
    W(x){
        dig[++len]=x%10;
        x/=10;
    }
    int ans=0;
    UP(i,1,len+1){
        ans+=dfs(len,i,0,true);
    }
    return ans-(len-1);
}

main(){
    int kase;
    scanf("%lld",&kase);
    MEM(dp,-1);
    W(kase--){
        int a,b;
        scanf("%lld%lld",&a,&b);
//        cout<<solve(b)<<" "<<solve(a-1)<<endl;
        printf("%lld\n",solve(b)-solve(a-1) );
    }
}

/*
5
1 1000000000000000000
*/