【数位DP】杠杆数

题目

题目描述
如果把一个数的某一位当成支点，且左边的数字到这个点的力矩和等于右边的数字到这个点的力矩和，那么这个数就可以被叫成杠杆数。

比如4139就是杠杆数，把3当成支点，我们有这样的等式：4 * 2 + 1 * 1 = 9 * 1。

给定区间[x,y]，求出在[x,y]中有几个杠杆数。

输入格式
两个数，表示x,y。

输出格式
一个输出，表示区间[x,y]中杠杆数的个数。

输入输出样例
输入 #1 复制
7604 24324
输出 #1 复制
897
说明/提示
对于40%的数据，x<=y<=x+100000

对于100%的数据，1<=x<=y<=10^18

思路

数位DP
我们可以枚举支点的位置，对于每个满足条件的数，它所对应的支点是唯一的，原因是如果将支点右移，左边减去右边的差将严格单调增加。state表示力矩和（支点左边加支点右边），所以当state<0时，当前这个数不满足以i为支点成为杠杆数的情况，返回0。但当state==0时并不能就ans++了，因为当前枚举的位置可能还没枚举完。
枚举好支点，问题就转化为：求[1,x]中，以第i位为支点的杠杆数的个数。

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=20;
ll f[N][N][3000];
int a[N];
ll dfs(int pos,int p,int s,bool bz)
{
	if(pos==0) return s==0;
	if(s<0) return 0;
	if(!bz&&f[pos][p][s]!=-1) return f[pos][p][s];
	int r=bz?a[pos]:9; ll yjy=0;
	for(int i=0; i<=r; i++) yjy+=dfs(pos-1,p,s+i*(pos-p),bz&&(i==r));
	if(!bz) f[pos][p][s]=yjy;
	return yjy;
}
ll solve(ll x)
{
	int len=0;
	while(x)
	{
		a[++len]=x%10; x/=10;
	}
	ll ans=0;
	for(int i=1; i<=len; i++) ans+=dfs(len,i,0,1);
	return ans-len+1;
}
int main()
{
	ll l,r;
	scanf("%lld%lld",&l,&r);
	memset(f,-1,sizeof(f));
	printf("%lld",solve(r)-solve(l-1));
}