bzoj1833 count 数字计数

传送门

常规数位$DP$。模板写起来真的很舒服。
统计每个数字出现次数。转化为：
数字$i$出现$j$次的数的个数。$0$需要特殊处理，因为要考虑前导$0$。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int a[20],num;
ll g[20][20][2][2],f[20][20][2],l,r,w[10];
inline ll dp(int pos,int res,bool limit,ll ans=0){
	if(!pos) return !res;
	if(~f[pos][res][limit]) return f[pos][res][limit];
	int up=limit?a[pos]:9;
	for(int i=0;i<=up;++i) ans+=dp(pos-1,res-(i==num),limit&&(i==up));
	return f[pos][res][limit]=ans;
}
inline ll dp0(int pos,int res,bool limit,bool lead,ll ans=0){
	if(!pos) return !res;
	if(~g[pos][res][limit][lead]) return g[pos][res][limit][lead];
	int up=limit?a[pos]:9;
	for(int i=0;i<=up;++i) ans+=dp0(pos-1,res-((!lead)&&(!i)),limit&&(i==up),lead&&(!i));
	return g[pos][res][limit][lead]=ans;
}
inline void solve(ll x,int id,int pos=0){
	while(x) a[++pos]=x%10,x/=10;
	memset(g,-1,sizeof(g));
	for(int i=1;i<=pos;++i) w[0]+=1ll*i*dp0(pos,i,true,true)*id;
	for(num=1;num<=9;++num){
		memset(f,-1,sizeof(f));
		for(int i=1;i<=pos;++i) w[num]+=1ll*i*dp(pos,i,true)*id;
	}
}
int main(){
	cin>>l>>r;
	solve(r,1),solve(l-1,-1);
	for(int i=0;i<=9;++i) cout<<w[i]<<' ';
}