【ybt高效进阶5-3-3】【luogu P2602】数字计数（数位DP）

数字计数

题目链接：ybt高效进阶5-3-3 / luogu P2602

题目大意

问你 l~r 的数中，每个数码出现了多少次。

思路

看到这种题不难想到数位 DP。

首先把求 $l\sim r$ 转为 $1\sim r$ 的结果减去 $1\sim l-1$ 的。

考虑从高位向低位 DP。
如果当前为 $i$，那如果选 $0\sim i-1$，就后面都是随便选，这个每个数的贡献可以提前预处理 DP 出来。
（不管前导 $0$ 会发现每个数的数码出现次数都一样）
（简单一算得到 $nu{m}_i=nu{m}_{i-1}\ast 10+10^{i-1}$）

然后如果这个数选 $i$，那后面就不能随便算，就要继续看下一位。
那在后面中能有多少个数可以算出来，那答案就加上多少个 $i$。

接着我们来考虑前面不管的前导 $0$。
会发现如果有 $i$ 位，前导 $0$ 可能是 $1,2,...,i$ 个。
对于每种这样的，它的个数分别是 $2^{i-1},2^{i-2},...,0$。
然后你算一算要第 $i$ 位成为前导 $0$ 有多少可能，每个都拿答案减去就行了。

代码

#include<cstdio>
#define ll long long

using namespace std;

ll l, r, ans[10];
ll num[13], ten[13];
int a[13], an;

void work(ll x, int op) {
	an = 0;
	while (x) {
		a[++an] = x % 10;
		x /= 10;
	}
	
	for (int i = an; i >= 1; i--) {
		for (int j = 0; j <= 9; j++)//没有到顶，后面随便选
			ans[j] += 1ll * op * a[i] * num[i - 1];
		for (int j = 0; j < a[i]; j++)
			ans[j] += 1ll * op * ten[i - 1];
		
		ll ainum = 0;//到顶，先计算这一位会出现的次数，然后待会再处理后面的位数
		for (int j = i - 1; j >= 1; j--) {
			ainum = ainum * 10ll + a[j];
		}
		ans[a[i]] += 1ll * op * (ainum + 1);
	}
	for (int i = an - 1; i >= 0; i--) ans[0] -= 1ll * op * ten[i];
	//处理前导0
}

int main() {
	ten[0] = 1;
	for (int i = 1; i <= 12; i++)
		ten[i] = ten[i - 1] * 10ll;
	for (int i = 1; i <= 12; i++) {
		num[i] = num[i - 1] * 10ll + ten[i - 1];
	}
	
	scanf("%lld %lld", &l, &r);
	work(r, 1); work(l - 1, -1);
	
	for (int i = 0; i <= 9; i++) printf("%lld ", ans[i]);
	
	return 0;
}