【BZOJ】1833: [ZJOI2010]count 数字计数(数位dp)

http://www.lydsy.com/JudgeOnline/problem.php?id=1833

数位dp什么的最恶心了。

dfs时注意考虑两种边界,一种是此时正好在这个数上,那么答案应该加的是后边的数+1+dfs

否则就加10^(x-1)+dfs;这两个是显然的。自己多想想就懂了

具体看代码:

#include <cstdio>

#include <cstring>

#include <cmath>

#include <string>

#include <iostream>

#include <algorithm>

#include <queue>

using namespace std;

typedef long long ll;

#define rep(i, n) for(int i=0; i<(n); ++i)

#define for1(i,a,n) for(int i=(a);i<=(n);++i)

#define for2(i,a,n) for(int i=(a);i<(n);++i)

#define for3(i,a,n) for(int i=(a);i>=(n);--i)

#define for4(i,a,n) for(int i=(a);i>(n);--i)

#define CC(i,a) memset(i,a,sizeof(i))

#define read(a) a=getint()

#define print(a) printf("%d", a)

#define dbg(x) cout << (#x) << " = " << (x) << endl

#define printarr2(a, b, c) for1(_, 1, b) { for1(__, 1, c) cout << a[_][__]; cout << endl; }

#define printarr1(a, b) for1(_, 1, b) cout << a[_] << '\t'; cout << endl

inline const ll getint() { ll r=0, k=1; char c=getchar(); for(; c<'0'||c>'9'; c=getchar()) if(c=='-') k=-1; for(; c>='0'&&c<='9'; c=getchar()) r=r*10+c-'0'; return k*r; }

inline const int max(const int &a, const int &b) { return a>b?a:b; }

inline const int min(const int &a, const int &b) { return a<b?a:b; }



ll f[100], c[100], a[100], p[100];



ll dfs(int x, int dig, int front, int line) {

	if(!x) return 0;

	if(!front && !line && f[x]!=-1) return f[x];

	ll last=(line?a[x]:9), tot=0;

	for1(i, 0, last) {

		if(front && i==0) tot+=dfs(x-1, dig, 1, line&&i==last);

		else if(i==dig) {

			if(i==last && line) tot+=c[x-1]+1+dfs(x-1, dig, 0, line&&i==last); //正好在这个数上

			else tot+=p[x-1]+dfs(x-1, dig, 0, line&&i==last);

		}

		else tot+=dfs(x-1, dig, 0, line&&i==last);

	}

	if(!front && !line) f[x]=tot;

	return tot;

}

ll getans(ll x, int dig) {

	CC(f, -1);

	ll t=x; int len=0;

	while(t) a[++len]=t%10, t/=10, c[len]=c[len-1]+a[len]*p[len-1];

	return dfs(len, dig, 1, 1);

}

int main() {

	ll a=getint(), b=getint();

	p[0]=1; for1(i, 1, 15) p[i]=p[i-1]*10;

	rep(i, 9) printf("%lld ", getans(b, i)-getans(a-1, i));

	printf("%lld\n", getans(b, 9)-getans(a-1, 9));

	return 0;

}

 

 


 

 

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。

Source

你可能感兴趣的:(count)