BZOJ 1833 数字计数（统计[a,b]每个数字出现次数）

题目链接：http://61.187.179.132/JudgeOnline/problem.php?id=1833

题意：给定区间[a,b]。求区间内0到9每个数字出现的次数。

思路：f[i][j]表示到后i位是否全 0（j=1表示i位之前全0）这个状态某个数字出现的次数，p[i][j]表示这个状态后面有多少个数字。那么当前枚举到的数字为要统计的数字时，答案加 上后面还有多少种数字，即下一个状态的p值。那么我们枚举要统计的数字依次统计即可。

i64 f[20][2],p[20][2];
i64 n,m;
int a[20],aNum;

int cur;

pair<i64,i64> DFS(int dep,int flag,int allZero)
{
if(dep==-1)  return MP(0,1);
if(!flag&&f[dep][allZero]!=-1) return MP(f[dep][allZero],p[dep][allZero]);
i64 x=0,y=0;
int L=flag?a[dep]:9;
int i;
for(i=0;i<=L;i++)
{
pair<i64,i64> q=DFS(dep-1,flag&&(i==L),allZero&&(i==0));
x+=q.first;
y+=q.second;
if(i==cur)
{
if(i==0&&!allZero||i!=0) x+=q.second;
}
}
if(!flag)
{
f[dep][allZero]=x;
p[dep][allZero]=y;
}
return MP(x,y);
}

i64 cal(i64 n,int x)
{
if(n==0) return 0;
cur=x; aNum=0;
while(n)
{
a[aNum++]=n%10;
n/=10;
}
pair<i64,i64> p=DFS(aNum-1,1,1);
return p.first;
}

int main()
{
RD(n); RD(m);
i64 a[20]={0};
int i;
FOR0(i,10)
{
clr(f,-1);
a[i]=cal(m,i)-cal(n-1,i);
}
printf("%lld",a[0]);
FOR1(i,9) printf(" %lld",a[i]);
puts("");
}