洛谷 P2602 [ZJOI2010]数字计数 数位DP（记忆化搜索）

P2602 [ZJOI2010]数字计数

题目描述

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

输入输出格式

输入格式：

 

输入文件中仅包含一行两个整数a、b，含义如上所述。

 

输出格式：

 

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

 

输入输出样例

输入样例#1： 复制

1 99

输出样例#1： 复制

9 20 20 20 20 20 20 20 20 20

说明

30%的数据中，a<=b<=10^6；

100%的数据中，a<=b<=10^12。

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
#include<map>
#include<set>
#include<cmath>
#include<cctype>
#include<ctime>
#define INF 0x3f3f3f3f
#define PI acos(-1.0)
#define MOD 998244353
typedef long long ll;
typedef unsigned long long ull;
const int mm=100000;

using namespace std;

ll a[15],dp[15][15][2],x;

ll dfs(int len,int pos,bool cnt,bool limit) //len当前数字长度 pos想要找的数字出现了几次 cnt是否有前导零 limit是否能达到最大值 
{
	if(len==-1) return (ll)pos;
	if(!limit&&dp[len][pos][cnt]!=-1) return dp[len][pos][cnt];
	int up=limit?a[len]:9;
	ll ans=0;
	for(int i=0;i<=up;i++)
	{
		ans+=dfs(len-1,pos+((i||cnt)&&(i==x)),i||cnt,i==a[len]&&limit);
	}
	if(!limit) dp[len][pos][cnt]=ans;
	return ans;
}

ll fun(ll x)
{
	int len=0;
	while(x)
	{
		a[len++]=x%10;
		x/=10;
	}
	return dfs(len-1,0,0,1);
}

int main()
{
	ll n,m;
	cin>>n>>m;
	while(x<10)
	{
		memset(dp,-1,sizeof(dp));
		cout<<fun(m)-fun(n-1)<<' ';
		x++;
	}
	return 0;
}