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

}