hdu 3709 数位dp

时间:2021-06-24 15:10:35

数位dp,有了进一步的了解,模板也可以优化一下了

题意:找出区间内平衡数的个数,所谓的平衡数,就是以这个数字的某一位为支点,另外两边的数字大小乘以力矩之和相等,即为平衡数
例如4139,以3为支点
4*2 + 1*1 = 9 and 9*1 = 9,称为平衡数

Sample Input
2
0 9
7604 24324

Sample Output
10
897

 #include<cstdio>
#include<cstring>
using namespace std;
__int64 dp[][][];
int digit[];
__int64 dfs(int p,int o,int s,bool e) { //数位,支点位置,前面到支点的和,是否任意填
if (p==-) return s==;
if(s<) return ;
if (!e &&dp[p][o][s]!=-) return dp[p][o][s];
__int64 res = ;
int u = e?digit[p]:;
for (int d=;d<=u;++d)
{
int ns=d*(p-o)+s;
res+=dfs(p-,o,ns,e&&d==u);
}
return e?res:dp[p][o][s]=res;
}
__int64 solve(__int64 n)
{
int len=;
while(n)
{
digit[len++]=n%;
n/=;
}
__int64 ans=;
for(int i=;i<len;i++)
{
ans+=dfs(len-,i,,);
}
return ans-len+;
}
int main()
{
int t;
__int64 n,m;
//freopen("1.in","r",stdin);
memset(dp,-,sizeof(dp));
scanf("%d",&t);
while(t--)
{
scanf("%I64d%I64d",&n,&m);
printf("%I64d\n",solve(m)-solve(n-));
}
}