首页 > 代码库 > BZOJ 1833 ZJOI 2010 count 数字计数 数位DP

BZOJ 1833 ZJOI 2010 count 数字计数 数位DP

题目大意:问0~9这10个数字在[l,r]中出现过多少次。


思路:数位DP。以前只是听说过,并没有写过,写了才发现好闹心啊。。

预处理一个数组,f[i][j][k]表示长度为i,开头为j,数字k出现的次数。

对于一个数kXXXXXX,我们先处理1~999999,然后处理1000000~kXXXXXX

前面的东西很规则,可以直接调用f数组来解决。

对于后面不太规则的东西,按位处理。总之就是乱搞,我也不太懂说不明白。。。


CODE:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

long long f[15][10][10],power[15];
long long a[10],b[10];

void Work(long long x,long long arr[])
{
	long long temp = x;
	int len = 0;
	static int digit[15];
	memset(digit,0,sizeof(digit));
	while(x)	digit[++len] = x % 10,x /= 10;
	x = temp;
	for(int i = 1; i < len; ++i)
		for(int j = 1; j <= 9; ++j)
			for(int k = 0; k <= 9; ++k)
				arr[k] += f[i][j][k];
	for(int i = len; i; --i) {
		for(int j = 0; j < digit[i]; ++j) {
			if(!j && i == len)	continue;
			for(int k = 0; k <= 9; ++k)
				arr[k] += f[i][j][k];
		}
		arr[digit[i]] += x % power[i] + 1;
	}
}

int main()
{
	power[1] = 1;
	for(int i = 2; i <= 12; ++i)
		power[i] = power[i - 1] * 10;
	for(int i = 0; i <= 9; ++i)
		f[1][i][i] = 1;
	for(int i = 2; i <= 12; ++i)
		for(int j = 0; j <= 9; ++j)
			for(int k = 0; k <= 9; ++k) {
				for(int z = 0; z <= 9; ++z)
					f[i][k][z] += f[i - 1][j][z];
				f[i][k][k] += power[i - 1];
			}
	long long l,r;
	scanf("%lld%lld",&l,&r);
	Work(r,a),Work(l - 1,b);
	for(int i = 0 ; i <= 9; ++i)
		printf("%lld%c",a[i] - b[i]," \n"[i == 9]);
	return 0;
}


BZOJ 1833 ZJOI 2010 count 数字计数 数位DP