洛谷 P4127 [AHOI2009]同类分布 (数位dp)

题目描述
给出两个数a,ba,b，求出[a,b][a,b]中各位数字之和能整除原数的数的个数。

输入格式
一行，两个整数aa和bb

输出格式
一个整数，表示答案

输入输出样例
输入 #1 复制
10 19
输出 #1 复制
3
说明/提示
对于所有的数据，1 ≤ a ≤ b ≤ 10^{18}1≤a≤b≤10
18

分析：

dp[i][j][k]:i表示数位，j表示到当前数位的各位数字之和，k表示原数。
由于原数很大，所以在计算原数的时候就对数字之和取模，那就要提前设置一个数字之和作为模数，所以枚举所有可能的个位数字之和。
优化：
1.当前的sum如果之后数位均为9还是小于模数，则return 0
2.如果当前的sum+i>mod，就跳过

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e3+10;
const int mod=1e9+7;
int a[20],fx;
ll dp[20][200][200];
ll dfs(int pos,int flag,int sum,int st)
{
	if(sum+9*pos<fx) return 0;
	if(pos<1) 
	{
		if(sum==fx&&st==0) return 1;
		else return 0;
	}
	if(!flag&&dp[pos][sum][st]!=-1) return dp[pos][sum][st];
	int up=flag?a[pos]:9;
	ll ans=0;
	for(int i=0;i<=up;i++)
	{
		if(sum+i<=fx)
		{
			ans+=dfs(pos-1,flag&&i==a[pos],sum+i,(st*10+i)%fx);
		}
		
	}
	if(!flag) dp[pos][sum][st]=ans;
	return ans;
}
ll solve(ll x)
{
	int k=0;
	while(x)
	{
		a[++k]=x%10;
		x=x/10;
	}
	ll ans=0;
	for(int i=1;i<=9*k;i++)
	{
		fx=i;
		memset(dp,-1,sizeof(dp));
		ans+=dfs(k,1,0,0);
	}
	return ans;
}
int main()
{
	ll a,b;
	scanf("%lld%lld",&a,&b);
	printf("%lld",solve(b)-solve(a-1));
	return 0;
}