【bzoj1833】 ZJOI2010—count 数字计数

http://www.lydsy.com/JudgeOnline/problem.php?id=1833 (题目链接)

题意

　　求在${[a,b]}$范围内整数中，每个数码出现的次数。

Solution

　　数位dp。

　　${t}$数组取到最大数时表示每一位是多少。

　　${f[i][j][k]}$表示第${i}$位，这一位上的数为${j}$，数字${k}$的出现次数。转移：$${f[i][j][k]=sum_{l=0}^9f[i-1][l][k]+10^(i-1)}$$

　　${g[i][k]}$表示第${i}$位，这一位上的数取到最大时数字${k}$的出现次数。转移：$${g[i][k]=sum_{j=0}^{t[i-1]-1}f[i-1][j][k]+g[i-1][k]}$$

　　当然，如果${g[i][t[i]]}$还要再加上${t[i]}$对方案的贡献。

细节

　　感觉细节还是蛮多的，还是不够熟练啊

代码

// bzoj1833
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<ctime>
#define LL long long
#define inf (1ll<<30)
#define MOD 1000000007
#define Pi acos(-1.0)
#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
using namespace std;

LL a[2],t[20],f[20][10][10],g[20][10],ans[10];
int n;

void solve(int p) {
	memset(f,0,sizeof(f));memset(g,0,sizeof(g));
	for (n=0;a[p];a[p]/=10) t[++n]=a[p]%10;
	for (int i=0;i<10;i++) f[1][i][i]=1;
	LL bin=1,bb=0;
	g[1][t[1]]=1;
	for (int i=2;i<=n;i++) {
		bb+=bin*t[i-1],bin*=10;
		for (int j=0;j<10;j++) {
			for (int k=0;k<10;k++) 
				for (int l=0;l<10;l++) f[i][j][k]+=f[i-1][l][k];
			f[i][j][j]+=bin;
		}
		for (int k=0;k<10;k++) {
			g[i][k]=g[i-1][k];
			for (int j=0;j<t[i-1];j++) g[i][k]+=f[i-1][j][k];
		}
		g[i][t[i]]+=bb+1;
	}
	int q=p ? 1 : -1;
	for (int i=1;i<n;i++)
		for (int j=1;j<10;j++)
			for (int k=0;k<10;k++)
				ans[k]+=f[i][j][k]*q;
	for (int j=1;j<t[n];j++)
		for (int k=0;k<10;k++)
			ans[k]+=f[n][j][k]*q;
	for (int k=0;k<10;k++)
		ans[k]+=g[n][k]*q;
}
int main() {
	scanf("%lld%lld",&a[0],&a[1]);a[0]--;
	solve(0);solve(1);
	for (int i=0;i<10;i++) {
		printf("%lld",ans[i]);
		if (i<9) printf(" ");
	}
	return 0;
}