用记忆化搜索解决数位DP的套路总结

$$\mathrm{P}\mathrm{r}\mathrm{o}\mathrm{b}\mathrm{l}\mathrm{e}\mathrm{m}$$

$\mathrm{P}\mathrm{r}\mathrm{o}\mathrm{b}\mathrm{l}\mathrm{e}\mathrm{m}1$

区间[L,R]中每个数码出现的次数。

• The Counting Problem——多组询问
• ZJOI2007 数字统计——单组询问

$\mathrm{P}\mathrm{r}\mathrm{o}\mathrm{b}\mathrm{l}\mathrm{e}\mathrm{m}2$

求区间[L,R]中，在二进制下0的个数比1多的有多少个。

• Round Numbers——From USACO

$\mathrm{P}\mathrm{r}\mathrm{o}\mathrm{b}\mathrm{l}\mathrm{e}\mathrm{m}3$

求区间[L,R]中，有多少个数字能够被各个数位的数字和整除。

• AHOI2009 同类分布——单组询问
• BZOH3003 月之谜——多组询问

$$\mathrm{S}\mathrm{o}\mathrm{l}\mathrm{u}\mathrm{t}\mathrm{i}\mathrm{o}\mathrm{n}$$

有关数位DP一般可以将$[L,R]$的答案转换成$[1,R]$减去$[1,L-1]$!的答案。

在枚举数字的时候，我们常见的参数有两个：$\mathrm{i}\mathrm{s}\mathrm{m}\mathrm{i}\mathrm{n}$和$\mathrm{z}\mathrm{e}\mathrm{r}\mathrm{o}$.前者表示枚举数字的上街，例如数字$123$，第二个数字的上界如果是$2$，$\mathrm{i}\mathrm{s}\mathrm{m}\mathrm{i}\mathrm{n}=1$,否则为$0$，那么枚举的上界为9.同理，如果当前位存在前导$0$，$\mathrm{z}\mathrm{e}\mathrm{r}\mathrm{o}=1$,否则为$0$.

$\mathrm{i}\mathrm{s}\mathrm{m}\mathrm{i}\mathrm{n}$的更新：对于当前位的上界x，有$\mathrm{i}\mathrm{s}\mathrm{m}\mathrm{i}\mathrm{n}\mathrm{a}\mathrm{n}\mathrm{d}now=x$.

$\mathrm{z}\mathrm{e}\mathrm{r}\mathrm{o}$的更新，有$\mathrm{z}\mathrm{e}\mathrm{r}\mathrm{o}\mathrm{a}\mathrm{n}\mathrm{d}now=0$.

然后就是记忆化，我们知道记忆化的数组$f[x][ans]$表示还剩下$x$位，在$\mathrm{i}\mathrm{s}\mathrm{m}\mathrm{i}\mathrm{n}=0$时，答案参数为$ans$的答案。为什么要$\mathrm{i}\mathrm{s}\mathrm{m}\mathrm{i}\mathrm{n}=0$呢？因为在等于1的情况下不同的数字答案是不同的，但是在等于$0$的时候枚举的上界都是$9$，都具有共性。所以我们应该这样记忆化。数字很大，但是某一些情况下答案参数却很小。因此我们可以记录答案参数。

$\mathrm{S}\mathrm{o}\mathrm{l}\mathrm{u}\mathrm{t}\mathrm{i}\mathrm{o}\mathrm{n}1$

记录参数$\mathrm{z}\mathrm{e}\mathrm{r}\mathrm{o}$，$\mathrm{i}\mathrm{s}\mathrm{m}\mathrm{i}\mathrm{n}$，$x$，$sum$，$num$。

前两者用于更新枚举的新数，$x$表示还剩下几位，$sum$表示有多少个数字等于$num$。

那么这里的答案参数就是$sum$，用$f[zero][x][sum][num]$记录状态即可。

边界条件：当$x=0$时返回$sum$。

$\mathrm{S}\mathrm{o}\mathrm{l}\mathrm{u}\mathrm{t}\mathrm{i}\mathrm{o}\mathrm{n}2$

记录参数$\mathrm{z}\mathrm{e}\mathrm{r}\mathrm{o}$，$\mathrm{i}\mathrm{s}\mathrm{m}\mathrm{i}\mathrm{n}$，$x$，$num$.

$num$表示$0$的个数减去$1$的个数。

用$f[x][num][zero]$记录状态即可。

边界返回：$$\{\begin{array}{l}num\ge 0,1\\ nun<0,0\end{array}$$

$\mathrm{S}\mathrm{o}\mathrm{l}\mathrm{u}\mathrm{t}\mathrm{i}\mathrm{o}\mathrm{n}3$

如果对于只有一组询问的情况，我们只需要枚举每一个数字和P。

记录参数$\mathrm{z}\mathrm{e}\mathrm{r}\mathrm{o}$，$\mathrm{i}\mathrm{s}\mathrm{m}\mathrm{i}\mathrm{n}$，$x$，$sum1$，$sum2$.

$sum1$表示数字和，$sum2$表示这个数字对P取模得数值。

在边界返回：$$\{\begin{array}{l}sum1=P\mathrm{a}\mathrm{n}\mathrm{d}sum2=0,1\\ sum1≠P\mathrm{o}\mathrm{r}sum2≠0,0\end{array}$$

我们发现记忆化数组参数的上限是$f(x,sum1,sum2)$.否则会超出空间限制。

我们可以将所有询问离线，再枚举模数，对于相同模数时对每一个数字做一遍记忆化搜索即可。

$$\mathrm{C}\mathrm{o}\mathrm{d}\mathrm{e}$$

$\mathrm{C}\mathrm{o}\mathrm{d}\mathrm{e}1$

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

#define int long long

using namespace std;

int m;
int f[100][2][100][10], a[100]; 

int read(void)
{
	int s = 0, w = 0; char c = getchar();
	while (c < '0' or c > '9') w |= c == '-', c = getchar();
	while (c >= '0' and c <= '9') s = s*10+c-48, c = getchar();
	return w ? -s : s;
}

int dp(int x,int zero,int isMin,int sum,int num)
{
	if (!isMin and ~f[x][zero][sum][num]) 
		return f[x][zero][sum][num];
	if (x == 0) return sum;
	int cnt = 0;
	for (int i=0;i<10;++i)
	{
		if (isMin && i > a[x]) break;
		cnt += dp(x-1,zero&&i==0,isMin&&i==a[x],sum+((!zero||i)&&i==num),num);
	}
	if (!isMin) f[x][zero][sum][num] = cnt;
	return cnt;
}

int work(int x,int num)
{
	m = 0;
	while (x) {
		a[++m] = x % 10;
		x /= 10;
	}
	return dp(m,1,1,0,num);
}

signed main(void)
{
	freopen("test.in","r",stdin);
	freopen("test.out","w",stdout);
	int l = read(), r = read();
	memset(f,-1,sizeof f);
	while (l and r)
	{
		for (int i=0;i<10;++i) 
			printf("%lld ", work(r,i)-work(l-1,i));
		puts("");
		l = read(), r = read();
	}
	return 0;
}

$\mathrm{C}\mathrm{o}\mathrm{d}\mathrm{e}2$

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

#define int long long

using namespace std;

int a[200], m;
int f[200][2][2][200];

int dp(int x,int zero,int isMin,int sum)
{
	if (~f[x][zero][isMin][sum]) return f[x][zero][isMin][sum];
	if (x == m+1) return sum <= 0;
	int cnt = 0;
	for (int i=0;i<2;++i)
	{
		if (isMin and i > a[x]) continue;
		int t = 0;
		if (i == 1) t = 1;
		else if (zero == 0) t = -1;
		cnt += dp(x+1,zero&&i==0,isMin&&i==a[x],sum+t);
	}
	return f[x][zero][isMin][sum] = cnt;
}

int ask(int x)
{
	memset(a,0,sizeof a);
	memset(f,-1,sizeof f);
	m = 0;
	while (x) {
		a[++m] = x & 1;
		x >>= 1;
	}
	reverse(a+1,a+m+1);
	return dp(1,1,1,0);
}

signed main(void)
{
	freopen("rndnum.in","r",stdin);
	freopen("rndnum.out","w",stdout);
	int l, r;
	cin>>l>>r;
	//cout<<ask(r)<<' '<<ask(l-1)<<endl;
	cout<<ask(r)-ask(l-1)<<endl;
	return 0;
}

$\mathrm{C}\mathrm{o}\mathrm{d}\mathrm{e}3$

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

#define int long long

using namespace std;

int n = 0, P, m;
int num[100000], a[30], f[21][401][401], ans[100000];

//sum1表示各位数字之和
//sum2表示数字对P取模的和 
int dp(int x,int sum1,int sum2,int isMin)
{
	if (x == 0) return sum1 == P and sum2 == 0;
	if (sum1 + x * 9 < P) return 0;
	if (!isMin and ~f[x][sum1][sum2]) return f[x][sum1][sum2];
	int Max = isMin ? a[x] : 9, cnt = 0;
	for (int i=0;i<=Max;++i)
	    cnt += dp(x-1,sum1+i,(sum2*10+i)%P,isMin&&i==a[x]);
	if (!isMin) f[x][sum1][sum2] = cnt;
	return cnt;
}

int Solve(int x)
{
	m = 0;
	while (x > 0) {
		a[++m] = x % 10;
		x /= 10;
	}
	if (m * 9 < P) return 0;
	return dp(m,0,0,1);
}

signed main(void)
{
	freopen("mystery.in","r",stdin);
	freopen("mystery.out","w",stdout);
	int t1, t2;
	while (cin>>t1>>t2) 
		num[++n] = t1-1, num[++n] = t2;
	for (P=1;P<=20*9;++P)
	{
		memset(f,-1,sizeof f);
	    for (int i=1;i<=n;++i) ans[i] += Solve(num[i]);
	}
	for (int i=1;i<=n;i+=2) 
	    printf("%lld\n", ans[i+1]-ans[i]);
	return 0;
}