hdu3709 Balanced Number(数位dp 枚举)

题目

T(T<=30)组样例，每次给出[x,y](0<=x<=y<=1e18)，求[x,y]中的“平衡的数”的数量

“平衡的数”：若存在一个轴，使得左边的数位到这个轴的力*力臂之和=右边的力*力臂之和，

则称该数是平衡的，其中力定义为数位的值，力臂定义为到轴的距离，

如4139是平衡的，因为其存在轴3，使得4*2+1*1=9*1

思路来源

https://www.cnblogs.com/bin-gege/p/5696129.html

题解

考虑枚举轴，轴一旦确定，每一位到轴的贡献都是确定的，

只关注dif=左边-右边=0的情况，就把dif也代入一维，最坏情况9*17+9*16+...+9，和约为1300，

dp[i][j][k]表示第i位，当前轴为j，当前轴左边-轴右边的值为k时的方案数

设数十进制宽为len，则0会被当成0，00，000，…，共计入len次，应减去len-1次多余

只有0会被计重，不妨考虑当前左=右，

向右挪动轴的时候，只要存在非0部分，左边之和就大于右边之和了，故其他数的贡献仅一次

代码

#include<bits/stdc++.h>
using namespace std;
const int N=20,M=2e3+10;
typedef long long ll;
int t,k,len;
ll l,r,a[N],sz[N][N][M];//dp[i][j][k]:表示i位 轴为j 当前左边比右边和大k时的方案数
//dp[pos][now][sum]
ll dfs(int pos,int now,int sum,bool lim){
	if(pos==0){
        return !sum;
	}
	if(sum<0){//由于先递增再递减 一旦小于0即无解
        return 0;
	}
	if(~sz[pos][now][sum] && !lim)return sz[pos][now][sum];
    ll ret=0;
    int up=lim?a[pos]:9;
    for(int i=0;i<=up;++i){
        ret+=dfs(pos-1,now,sum+(pos-now)*i,lim && i==up);
    }
    if(!lim)sz[pos][now][sum]=ret;
    return ret;
}

ll part(ll x){
	len=0;
	while(x)a[++len]=x%10,x/=10;
	ll ans=0;
	//枚举轴为[1,len]
	for(int i=len;i>=1;--i){
        ans+=dfs(len,i,0,1);
	}
	return ans-(len-1);//0被记入len次 实际只应被计一次
}

int main(){
    memset(sz,-1,sizeof sz);
    scanf("%d",&t);
    for(int c=1;c<=t;++c){
        scanf("%lld%lld",&l,&r);
        if(l)printf("%lld\n",part(r)-part(l-1));
        else printf("%lld\n",part(r));
    }
	return 0;
}