BZOJ1833: [ZJOI2010]count 数字计数（洛谷P2602）

乱搞

BZOJ题目传送门
洛谷题目传送门

把题目转化成求1到n的出现次数。设$f[i]$表示第$i$位出现的总次数，那么有$f[i]=f[i-1]\ast 10+10^{i-1}$（先不管前导0）。

我们对n的每一位进行处理。对于第$i$位，没有到达边界的数字答案即为$10^{i-1}+f[i-1]\ast a[i]$，边界数字的答案为$10^{i-1}+n\%10^{i-1}$，超出边界的答案就是$10^{i-1}$。注意因为有前导0，0的答案要减去$\sum 10^{i-1}$。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 13
using namespace std;
typedef long long LL;
LL l,r,a[N],f[N],pw[N],ans[10][2];
inline void calc(LL x,int ff){
	memset(a,0,sizeof(a)); LL t=x;
	while (t) a[++a[0]]=t%10,t/=10;
	for (int i=a[0];i;i--){
		for (int j=0;j<=9;j++) ans[j][ff]+=f[i-1]*a[i];
		for (int j=0;j<a[i];j++) ans[j][ff]+=pw[i-1];
		ans[a[i]][ff]+=x%pw[i-1]+1,ans[0][ff]-=pw[i-1];
	}
}
int main(){
	scanf("%lld%lld",&l,&r),pw[0]=1;
	for (int i=1;i<13;i++) pw[i]=pw[i-1]*10;
	for (int i=1;i<13;i++) f[i]=f[i-1]*10+pw[i-1];
	calc(r,1),calc(l-1,0);
	for (int i=0;i<9;i++) printf("%lld ",ans[i][1]-ans[i][0]);
	return printf("%lld\n",ans[9][1]-ans[9][0]),0;	
}