Luogu_P4999 烦人的数学作业 数位DP 记搜

Luogu_P4999 烦人的数学作业

数位DP

---

题目链接
明显数位DP的题目
\(f[i][j]\)记为第\(i\)位，各位总和为\(j\)的数和。
记忆化搜索。

---

代码如下：

#include<bits/stdc++.h>
#define ll long long
#define int ll
using namespace std;
const ll mod=1e9+7;
int t;
ll a[20],f[20][200];
ll dfs(ll pos,ll sum,bool b){
	if(pos==0)	return sum%mod;
	if(!b && f[pos][sum]!=-1)  return f[pos][sum]%mod;
	ll maxx= b ? a[pos] : 9;ll ans=0;
	for(int i=0;i<=maxx;i++)
		ans=(ans+dfs(pos-1,sum+i,b && a[pos]==i))%mod;
	if(!b) f[pos][sum]=ans%mod;
	return ans%mod;
}
inline ll sol(ll x){
	ll tot=0;
	while(x>0){
		a[++tot]=x%10;x/=10;
	}
	return dfs(tot,0,1)%mod;
}
signed main()
{
	scanf("%d",&t);
	memset(f,-1,sizeof(f));
	while(t--){
		ll l,r;scanf("%lld%lld",&l,&r);
		printf("%lld\n",(sol(r)-sol(l-1)+mod)%mod);
	}
	return 0;
}