1833: [ZJOI2010]count 数字计数

数位DP裸题。

写代码5min,交上去,WA。。。。。。。。

对拍半小时,错误不断,终于过了。

果然对拍还是不能少,AC率又掉了55555555

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
ll f[20][20][20],ex[20];
void dp(){
	ex[0]=1;
	for(int i=1;i<=12;i++)ex[i]=ex[i-1]*10;
	for(int i=1;i<=12;i++)
	for(int j=0;j<=9;j++)
	for(int k=0;k<=9;k++){
		if(j==k)f[i][j][k]=ex[i-1];
		for(int h=0;h<=9;h++)
		f[i][j][k]+=f[i-1][h][k];
	}
}
ll calc(ll a,int k){
	if(!a)return 0;
	ll ans=0;
	int num[20],len=1;num[len]=a%10;a/=10;
	while(a){
		 num[++len]=a%10;
		 a/=10;
	}
	for(int i=1;i<num[len];i++)ans+=f[len][i][k];
	for(int i=len-1;i>=1;i--){
		for(int j=0;j<num[i];j++)
		ans+=f[i][j][k];
	}
	for(int i=len;i>1;i--)
	if(num[i]==k){
		ll x=0;
		for(int j=i-1;j>=1;j--)
		x=x*10+num[j];
		ans+=x+1;
	}
	if(num[1]==k)ans++;
	for(int i=1;i<len;i++){
		for(int j=1;j<=9;j++)
		ans+=f[i][j][k];
	}
	return ans;
}
int main(){
	ll a,b;scanf("%lld%lld",&a,&b);
	dp();
	for(int i=0;i<=9;i++)
	if(!i)printf("%lld",calc(b,i)-calc(a-1,i));
	else printf(" %lld",calc(b,i)-calc(a-1,i));
	return 0;
}


你可能感兴趣的:(1833: [ZJOI2010]count 数字计数)