bzoj 1833: [ZJOI2010]count 数字计数

Description

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

Input

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

Output

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

Sample Input

1 99

Sample Output

9 20 20 20 20 20 20 20 20 20

HINT

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

数位DP...f[i][k]表示前i位中k的个数...感觉并不需要记录首位

每次转移的时候枚举当前位，判断下是否等于k，等于的话再判断满足条件的长度为i-1的数有多少个。累加即可

#include<map>
#include<cmath>
#include<queue>
#include<vector>
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int a[15];
long long ans[2][11];
long long f[15][15];
inline long long con(int len,bool lim)
{
	if(lim)
	{
		long long p=a[len];
		int i;
		for(i=len-1;i>=1;i--)
			p=p*(long long)10+a[i];
		p++;
		return p;
	}
	else
	{
		long long p=1;
		int i;
		for(i=1;i<=len;i++)
			p=p*(long long)10;
		return p;
	}
}
inline long long dfs(int len,int k,bool lim,bool lim2)
{
	if(len==0)
		return 0;
	if(!lim&&lim2&&f[len][k]!=-1)
		return f[len][k];
	int limit;
	if(lim)
		limit=a[len];
	else
		limit=9;
	int i;
	long long ans=0;
	for(i=0;i<=limit;i++)
	{
		ans+=dfs(len-1,k,lim&&i==a[len],lim2||i!=0);
		if(i==k&&i!=0||i==0&&i==k&&lim2)
			ans+=con(len-1,lim&&i==a[len]);
	}
	if(!lim&&lim2)
		f[len][k]=ans;
	return ans;
}
inline void cale(long long x,int d)
{
	if(x<0)
		return ;
	int len=0;
	a[1]=0;
	while(x!=0)
	{
		len++;
		a[len]=x%(long long)10;
		x=x/(long long)10;
	}
	int i;
	for(i=0;i<=9;i++)
		ans[d][i]=dfs(len,i,true,false);
	ans[d][0]++;
}
int main()
{
	long long l,r;
	scanf("%lld%lld",&l,&r);
	memset(f,-1,sizeof(f));
	cale(r,0);
	cale(l-1,1);
	int i,j;
	for(i=0;i<=8;i++)
		printf("%lld ",ans[0][i]-ans[1][i]);
	printf("%lld\n",ans[0][i]-ans[1][i]);
	return 0;
}