HDU-3709 Balanced Number

数位DP

对于一个数，我们可以枚举它的中心点，判断它是不是一个平衡数

所以在数位DP一开始时就加一个个条件，以第m位为中心点

d[m][len][f]代表中心为m，长度为len，左边减右边差值+1800为f的数个数

最后判断下f是否为修正值1800

这样DP会重复一些0，最后要减掉

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int N=20;
const int fix=1800;
ll d[N][N][3600];
int dis[N],t;
ll dp(int m,int len,int f,bool flag)
{
	if(len==-1) return f==fix;
	if(!flag&&d[m][len][f]!=-1) return d[m][len][f];
	ll ans=0;
	int end=flag?dis[len]:9;
	for(int i=0;i<=end;i++)
		ans+=dp(m,len-1,f+(m-len)*i,flag&&i==end);
	if(!flag) d[m][len][f]=ans;
	return ans;
}
ll solve(ll n)
{
	if(n==-1) return 0;
	t=0;
	while(n)
	{
		dis[t++]=n%10;
		n/=10;
	}
	ll ans=0;
	for(int i=0;i<t;i++)
			ans+=dp(i,t-1,fix,true);
	return ans-t+1;
}
int main()
{
	int T;
	ll l,r;
	memset(d,-1,sizeof(d));
	scanf("%d",&T);
	while(T--)
	{
		scanf("%lld%lld",&l,&r);
		printf("%lld\n",solve(r)-solve(l-1));
	}
	return 0;
}