bzoj 1833

f[pos][top]

统计数字个数

g[pos][top][digit]

统计数位（对于f[pos][top]中的所有数）

/**************************************************************
    Problem: 1833
    User: idy002
    Language: C++
    Result: Accepted
    Time:0 ms
    Memory:808 kb
****************************************************************/
 
#include <cstdio>
#include <cstring>
 
typedef long long poi;
 
poi lf, rg;
int aa[20], tot;
poi f[20][2], g[20][2][10];
poi ans[2][10];
 
void calc( poi v, poi ans[10] ) {
    for( tot=0; v; v/=10 )
        aa[++tot] = v%10;
    memset( f, 0, sizeof(f) );
    memset( g, 0, sizeof(g) );
    f[tot][1] = 1;
    f[tot][0] = aa[tot]-1;
    for( int i=tot; i>=2; i-- ) {
        f[i-1][1] = 1;
        f[i-1][0] = f[i][0]*10+9+f[i][1]*aa[i-1];
    }
    g[tot][1][aa[tot]] = 1;
    for( int i=1; i<aa[tot]; i++ )
        g[tot][0][i] = 1;
    for( int i=tot; i>=2; i-- ) {
//      for( int s=0; s<=9; s++ )
//          g[i-1][0][s] = g[i][0][s]*10+f[i][0]+(s!=0);
        for( int s=0; s<=9; s++ ) {
            g[i-1][0][s] += f[i][0];
            for( int ss=0; ss<=9; ss++ ) {
                g[i-1][0][ss] += g[i][0][ss];
            }
        }
        for( int s=1; s<=9; s++ )
            g[i-1][0][s]++;
        for( int s=0; s<aa[i-1]; s++ ) {
            g[i-1][0][s] += f[i][1];
            for( int ss=0; ss<=9; ss++ ) {
                g[i-1][0][ss] += g[i][1][ss];
            }
        }
        for( int s=0; s<=9; s++ )
            g[i-1][1][s] = g[i][1][s]+(aa[i-1]==s);
    }
     
    for( int i=0; i<=9; i++ )
        ans[i] = g[1][0][i]+g[1][1][i];
}
 
int main() {
    scanf( "%lld%lld", &lf, &rg );
    calc(rg,ans[1]);
    if( lf>1 ) calc(lf-1,ans[0]);
    for( int i=0; i<=9; i++ )
        printf( "%lld%s", ans[1][i]-ans[0][i], i==9 ? "" : " " );
}