[省选前题目整理][BZOJ 1833][ZJOI 2010]count 数字计数(数位DP)

题目链接

思路

代码

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>

#define MAXN 1000

using namespace std;

typedef long long int LL;

LL ans[10]; //ans[i]=数位i在[L,R]中的出现次数
LL pow[13];

void solve(LL x,int sign) //sign为1在答案里加，sign为-1在答案里减
{
if(x==-1) //特判
{
ans[0]++;
return;
}
//求数字1~9的出现次数
for(LL j=1;j<=9;j++)
for(LL i=1;pow[i-1]<=x;i++) //枚举第i位
{
LL len=i-1; //len是该数字右边那部分的长度
LL L=x/pow[i],R=x%pow[i-1]; //L是该数字左边那部分，R是该数字右边那部分
LL t=(x/pow[i-1])%10; //t是x第i位的数位
if(j>t) ans[j]+=L*pow[len]*sign;
else if(j<t) ans[j]+=(L+1)*pow[len]*sign;
else if(j==t) ans[j]+=(L*pow[len]+R+1)*sign;
}
//求0的出现次数
for(LL i=1;pow[i-1]<=x;i++)
{
LL len=i-1; //len是该数字右边那部分的长度
LL L=x/pow[i],R=x%pow[i-1]; //L是该数字左边那部分，R是该数字右边那部分
LL t=(x/pow[i-1])%10; //t是x第i位的数位
if(t>0) ans[0]+=L*pow[len]*sign;
else if(t==0) ans[0]+=((L-1)*pow[len]+R+1)*sign;
}
}

int main()
{
LL a,b;
scanf("%lld%lld",&a,&b);
pow[0]=1;
for(int i=1;i<13;i++) pow[i]=pow[i-1]*10;
solve(b,1);
solve(a-1,-1);
for(int i=0;i<=9;i++) printf("%lld%c",ans[i],i==9?'\n':' ');
return 0;
}