J - Junior Mathematician(数位dp)

传送门

设$x$在十进制数下有$k$位,且$d(x,i)$表示$x$的第$i$位

定义$f(x)=\sum _{i=1}^{k-1}\sum _{j=i+1}^{k}d(x,i)\ast d(x,j)$

求在$[L,R]$内满足$x==f(x)\%m$的个数

---

考虑数位$dp$从高到低枚举每一个位置

定义$f[i][j][q][w]$为枚举到第$i$位,前缀大小为$j$,当前贡献为$q$,当前数字为$w$的方案数

考虑前$i-1$位前缀为$sum$,贡献为$res$,数字为$last$

那么第$i$位为$x$,当前位权值为$li$

前缀为$(sum+x)\%m$,贡献变为$(res+x\ast sum)\%m$,当前数字变为$(last+x\ast li)\%m$

这样时间空间复杂度都是$O(n{m}^3)=300000\ast 3600$

但是可以直接维护$f(x)-x$的值,因为我们只需要判断相等就好了

那么定义$f[i][j][q]$为枚举到第$i$位,前缀为$j$,$f(x)-x$为$q$的方案

设下一位填$x$

那么前缀变为$x+j$,$f(x)-x$变为$q+x\ast j-x\ast 10^i$

这题卡常太难受了…

#include <bits/stdc++.h>
using namespace std;
const int mod = 1e9+7;
int n,m,bit[5009],dig[5009],f[5009][61][61][2],id,ok[5009][61][61][2];
char a[5009],b[5009];
int dfs(int num,int pre,int fx,int limit)
{
	long long ans = 0;
	if( num==0 )	return fx==0;
	if( ok[num][pre][fx][limit]==id )	return f[num][pre][fx][limit];
	ok[num][pre][fx][limit] = id;
	int lim = limit?dig[num]:9;
	for(int i=0;i<=lim;i++)
	{
		int he = ((fx+i*pre-i*bit[num])%m+m)%m;
		ans = ans+dfs(num-1,(pre+i)%m,he,limit&&(i==dig[num]) );
	}
	ans = ( ans+mod )%mod;
	f[num][pre][fx][limit] = ans;
	return ans;
}
int solve( char a[] )
{
	++id;
	int n = strlen( a+1 );
	for(int i=1;i<=n;i++)	dig[i] = a[n-i+1]-'0';
	return dfs(n,0,0,1);	
}
int main()
{
	int t; cin >> t;
	while( t-- )
	{
		scanf("%s%s%d",a+1,b+1,&m);
		int n = strlen( a+1 ),n2 = strlen( b+1 );
		a[n]--;
		for(int i=n;i>=1;i--)
		{
			if( a[i]>='0'&&a[i]<='9' )	break;
			a[i] = '9', a[i-1]--;
		}
		bit[1] = 1;
		for(int i=2;i<=n2;i++)	bit[i] = bit[i-1]*10%m;
		cout << ( solve(b) - solve(a) + mod )%mod << endl;
	}
}