题解 P4999 【烦人的数学作业】

数位 dp。

设 $dp_{q,i}$（$i\in\{0,1,2,3,4,5,6,7,8,9\}$）为 $1\sim q$ 中 $i$ 出现的次数，$1\sim q$ 的数字和显然就是 $dp_{q,0}\times 0+dp_{q,1}\times 1+\cdots+dp_{q,i}\times i\cdots+dp_{q,9}\times 9$。

所以我们只需要求出 $1\sim q$ 中 $i$ 出现的次数就能解决这个问题了。

这个问题看起来很好解决，但是注意前导零会影响结果，所以不能有前导零。

这该怎么办呢？

有前导零的式子很容易推出。有 $q$ 位数字，$i$ 数码的出现次数对于 $x\in\{s\mid s\in \mathbb N,10^q\le s\le10^{q+1}\}$ ，$f(q,i)$ 的数量都是相等的（设 $f(q,i)$ 为 $q$ 位数 $i$ 数码的出现次数）。

具体求法罢，是：
$\begin{cases}f(q,i)=0&q=0\\f(q,i)=10f(q-1)+10^{q-1}&q>0\end{cases}$

我们考虑减去多余的 $0$。

我们先设数字为 $\overline{A_1A_2A_3\dots A_n}$

我们首先考虑求 $\overline{A_100\dots 0}$，将 $\overline{A_100\dots 0}$ 分割为区间 $[0000,1000),[1000,2000),\dots,[\overline{(A_1-1)00\dots 0},\overline{A_100\dots 0})$，所以答案就为 $10^{n-1}A_1$，注意 $<A_1$ 的每个数还出现了 $10^{n-1}$ 次，所以要加上。

首位 $A_1$ 出现了 $\overline{A_2A_3\dots A_n}+1$ 次，答案还要加上 $\overline{A_2A_3\dots A_n}+1$，

当然还需要处理前导 $0$，用排列组合算一下会知道 $i$ 位 $q$ 个前导零的数量就是 $10^q$（$q\in\{s\mid s\in\mathbb N,0\le s\le i-1\}$），把它们加起来会发现一共出现了 $10^{i-1}+10^{i-2}+...10$（$=\sum\limits_{k=0}^{i-1}10^k$） 次，减一下即可。

Code:

#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
const int N=51,MOD=1e9+7; //注意能 MOD 的地方都要 MOD，不然会 WA 0pts。
ll pow10[N],dp[N],a[N],count[N],tmpcount[N],ans;
// pow10    : 字面意思，10^n
// dp       : 不考虑前导零的状况
// count    : 统计 0~9 出现次数
// tmpcount : 暂时保存 count，用来减
// ans      : 累加答案
void init() //预处理 pow10 和 dp。
{
	pow10[0]=1;
	for (int i=1;i<30;i++) dp[i]=(dp[i-1]*10%MOD+pow10[i-1])%MOD,pow10[i]=10*pow10[i-1]%MOD;
}
void solve(ll x)
{
	int len=0;
	while (x){a[++len]=x%10;x/=10;} //数位分离
	for (int i=len;i>=1;i--)        //从高到低遍历
	{
		for (int j=0;j<10;j++) count[j]+=dp[i-1]*a[i],count[j]%=MOD;  //分割区间
		for (int j=0;j<a[i];j++) count[j]+=pow10[i-1],count[j]%=MOD; //加上 10^(n-1)
		ll lastnum=0;
		for (int j=i-1;j>=1;j--) lastnum=lastnum*10+a[j],lastnum%=MOD; //求出 A2A3A4...An
		count[a[i]]+=lastnum+1,count[a[i]]%=MOD;
		count[0]-=pow10[i-1],count[0]=(count[0]+MOD)%MOD; //减去前导零
	}
}
int main()
{
	init();
	ll l,r,T;
	cin>>T;
	for (int q=0;q<T;q++)
	{
		ans=0; cin>>l>>r;
		solve(r); //前缀和思想相减 r 和 l-1。
		for (int i=0;i<10;i++) (tmpcount[i]=count[i]),count[i]=0; //复制 count，记得清零
		solve(l-1);
		for (int i=0;i<10;i++) ans=(ans+i*(tmpcount[i]-count[i]+MOD)%MOD)%MOD,count[i]=0; //累加答案，记得清零 count。
		cout<<ans<<'\n';
	}
	return 0;
}

Refence 求数字 $i$ 出现的次数。