hdu_3709_Balanced Number(数位DP)

时间:2023-03-08 17:19:35

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

题意:给你一个区间,让你找平衡数的个数

题解:设dp[i][j][k]为前i位以第j位为支撑点的力矩和为k的方案数,注意的是0,00,000这些也是平衡数,所以要减掉一个len长度

 #include<cstdio>
#include<cstring>
#define F(i,a,b) for(LL i=a;i<=b;i++)
typedef long long LL; LL t,n,m,dp[][][],dig[],len,an,an2,ii; LL dfs(int pos,int now,int f,bool inf){
if(!pos)return !f;
if(f<)return ;
if(!inf&&dp[pos][now][f]!=-)return dp[pos][now][f];
LL end=inf?dig[pos]:,ans=;
F(i,,end)ans+=dfs(pos-,now,f+(pos-now)*i,inf&&(i==end));
if(!inf)dp[pos][now][f]=ans;
return ans;
} int main(){
memset(dp,-,sizeof(dp));
scanf("%lld",&t);
while(t--){
scanf("%lld%lld",&n,&m),n--;
for(len=;n;n/=)dig[++len]=n%;
for(an=,ii=len;ii>;ii--)an+=dfs(len,ii,,);an-=len;
for(len=;m;m/=)dig[++len]=m%;
for(an2=,ii=len;ii>;ii--)an2+=dfs(len,ii,,);
printf("%lld\n",an2-len-an);
}
return ;
}