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("");
}