HDU 3709 Balanced Number(数位DP)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3709

 

题目大意：如果一个数字选择一个数作为轴，这个被选中数字的左边的价值等于左边每个数位*到选中数字的距离加和与右边的价值(等于由边每个数位*到选中数字的距离加)，称为一个平衡数，问区间内有多少平衡数

 

题目思路：若一个数字是一个平衡树，那么其一定只有一个轴，因为假设其中一个数字是它的平衡轴，那么这个位置继续往右的左边部分一定变大，而右边顶多保持不减，所以右边一定不存在

 

有了这个结论就可以大胆的统计枚举每一位作为平衡轴的平衡数个数就行

 

考虑dp，直接就记录一路走下来看到最后值是否为0就行，假设位置，轴，值都相同，那么后面的情况一定相同，所以直接用这个作为dp的状态记录下来就行

 

以下是代码：

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define rep(i,a,b) for(ll i=a;i<=b;i++)
#define per(i,a,b) for(ll i=a;i>=b;i--)
#define ll long long
const ll MAXN=20+5;
const ll MOD =1e9+7;
ll x,y,a[MAXN],dp[MAXN][MAXN][2000];
ll dfs(ll pos,ll zhou,ll val,ll limit){
    if(!pos) return val==0;
    if(!limit&&dp[pos][zhou][val]!=-1)return dp[pos][zhou][val];
    ll up=limit?a[pos]:9,ans=0;
    rep(i,0,up){
        ans+=dfs(pos-1,zhou,val+(pos-zhou)*i,limit&&i==a[pos]);
    }
    if(!limit)dp[pos][zhou][val]=ans;
    return ans;
}
ll solve(ll x){
    if(x==-1)return 0;
    ll pos=0;
    while(x){
        a[++pos]=x%10;
        x/=10;
    }
    ll ans=0;
    rep(i,1,pos){
        ans+=dfs(pos,i,0,1);
    }
    return ans-(pos-1);
}
int main(){
    ll t;
    scanf("%lld",&t);
    memset(dp,-1,sizeof(dp));
    while(t--){
        scanf("%lld%lld",&x,&y);
        cout<<solve(y)-solve(x-1)<<endl;
    }
    return 0;
}