[多次数位dp]数字计数 洛谷P2602

题目描述

给定两个正整数 a 和 b，求在 [a,b] 中的所有整数中，每个数码(digit)各出现了多少次。

输入格式

仅包含一行两个整数 a,b，含义如上所述。

输出格式

包含一行十个整数，分别表示 0∼9 在 [a,b] 中出现了多少次。

输入输出样例

输入 #1

1 99

输出 #1

9 20 20 20 20 20 20 20 20 20

说明/提示

数据规模与约定

• 对于 30% 的数据，保证 a≤b≤10^6；
• 对于 100% 的数据，保证 1≤a≤b≤10^12。

题意： 求[a, b]的所有整数中，各个数码分别出现了多少次。

分析： 有两种做法，一种是在dfs过程中求出dp数组，一种是提前预处理出来dp数组，两种方法得到的数组含义不同，第一种的dp数组是dp[pos][cnt]，表示在搜到第pos位时前面有cnt个待求数字时后pos位任取的情况下出现的次数，第二种的dp[pos]表示某个数字在后pos位任取的情况下出现的个数，这里不考虑第pos位前的情况。由于第一种做法比较基础，这里就只记录一下第二种做法。

首先需要提前处理出来dp[pos]数组，由于0~9任何一个数字的dp值都是相同的，这里就只记录一次，后pos位任取时一共有10^pos的不同的数字，每个数字都是pos位的，又由于0~9每个数字出现概率相同，所以dp[pos] = pos*10^pos / 10，得到这个数组后在dfs时，如果当前位可以任取且前一位不是前导零，设cnt为目前已经搜到的待求数字个数，这种情况下的答案就是10^pos*cnt + dp[pos]，因为后pos位任取时会产生10^pos个数字，每个数字都会产生cnt个贡献，再加上后pos位的贡献就是当前情况下的总贡献。

具体代码如下：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
const int N=23;
ll f[N];
int a[N];
ll dfs(int pos,int cnt,int lead,int limit,int t)
{
	if(!pos)
	{
		if(lead&&t==0) return 1;//如果只有一个0，那么这个0也不能算是前导0 
		return cnt;
	}
	if(!lead&&!limit)
		return pow(10, pos)*cnt+f[pos];
	ll ans=0;
	int up=limit?a[pos]:9;
	for(int i=0;i<=up;i++)
	{
		if(!t) ans+=dfs(pos-1,cnt+(!lead&&(i==0)),lead&&(i==0),limit&&(i==up),t);
		else ans+=dfs(pos-1,cnt+(i==t),lead&&(i==0),limit&&(i==up),t);
	}
	return ans;
}
ll solve(ll x,int t)
{
	if(!x)
	{
		if(!t) return 1;
		return 0;
	}
	int pos=0;
	while(x)
	{
		a[++pos]=x%10;
		x/=10;
	}
	return dfs(pos,0,1,1,t);
}
int main()
{
	f[1] = 1;
	for(int i=2;i<=15;i++)
		f[i]=f[i-1]/(i-1)*i*10;
	ll l,r;
	cin>>l >> r;
	for(int i=0;i<=9;i++)
		cout<<solve(r,i)-solve(l-1,i)<<' ';
	return 0;
}