【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;

}

巴特西

【bzoj1833】 ZJOI2010—count 数字计数

题意

Solution

细节

代码

最新文章

热门文章