bzoj1833: [ZJOI2010]count 数字计数

链接

　　http://www.lydsy.com/JudgeOnline/problem.php?id=1833

题解

　　sth说0的统计要用容斥，但我觉得不需要啊。
　　f[i][j]表示没有限制的情况下，i位数字，数码j的出现次数。g[i][j]表示有限制的情况下，i为数字数码j的出现次数。（都允许有前导0）
　　转移水，主要是答案统计。对于ans[1]~ans[9]，直接ans[j]=g[位数][j]。对于0，因为有前导0的问题，先枚举这个数有多少位，然后从1到9(或者num[i])枚举最高位，统计答案就好了。

代码

//数位dp 
#include <cstdio>
#include <algorithm>
#define ll long long
using namespace std;
ll ans[10];
void work(ll x, ll y)
{
    ll num[20]={0}, N=0, f[20][10]={0}, g[20][10]={0}, i, j, mi[20], now, t;
    for(t=x;t;t/=10)num[++N]=t%10;
    for(i=mi[0]=1;i<=N;i++)mi[i]=mi[i-1]*10;
    for(i=1;i<=N;i++)
        for(j=0;j<=9;j++)
        {
            for(now=0;now<=9;now++)
            {
                f[i][j]+=f[i-1][j];
                if(now==j)f[i][j]+=mi[i-1];
            }
            for(now=0;now<=num[i];now++)
            {
                if(now<num[i])g[i][j]+=f[i-1][j];
                else g[i][j]+=g[i-1][j];
                if(now==j)
                {
                    if(now<num[i])g[i][j]+=mi[i-1];
                    else g[i][j]+=x%mi[i-1]+1;
                }
            }
        }
    for(i=1;i<=9;i++)ans[i]+=y*g[N][i];
    ans[0]+=y*g[N-1][0];
    for(i=1;i<num[N];i++)ans[0]+=y*f[N-1][0];
    for(i=1;i<N;i++)for(j=1;j<=9;j++)ans[0]+=y*f[i-1][0];
}
int main()
{
    ll a, b;
    scanf("%lld%lld",&a,&b);
    work(a-1,-1);
    work(b,1);
    printf("%lld",ans[0]);
    for(int i=1;i<=9;i++)printf(" %lld",ans[i]);
    return 0;
}