BZOJ-1833: [ZJOI2010]count 数字计数(数位统计)

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

之前用数位DP傻叉的搞了半天一直WA,今天重新写了一下,这里有个DFS的方法很不错,也很好写:http://hi.baidu.com/cenyk1230/item/4f1761767d75870cd0dcb3e6,然后我终于把恶心的数位统计A了一题额。。。

代码:

#include 
#include 
#include 
 
using namespace std ;
 
#define rep( i , m ) for ( int i = 0 ; i < m ; ++ i )
#define maxn 20
 
typedef long long ll ;
 
ll a , b , cnt[ maxn ] , ans[ maxn ] ;
int N[ maxn ] , n ; 
 
void dfs( int now ) {
    if ( ! now ) {
        rep( i , N[ now ] + 1 ) ++ cnt[ i ] ;
        return ; 
    }
    ll ret = 1 ; 
    rep( i , now ) ret *= 10 ;
    rep( i , N[ now ] ) {
        rep( j , 10 ) {
            cnt[ j ] += ( ret * now ) / 10 ;
        }
        cnt[ i ] += ret ;
    }
    ll rec = 0 ;
    for ( int i = now - 1 ; i >= 0 ; -- i ) {
        rec *= 10 , rec += N[ i ] ;
    }
    cnt[ N[ now ] ] += ( rec + 1 ) ;
    dfs( now - 1 ) ;
}
 
void cal( ll x ) {
    memset( cnt , 0 , sizeof( cnt ) ) ;
    for ( n = 0 ; x ; x /= 10 ) N[ n ++ ] = x % 10 ;
    if ( n ) dfs( n - 1 ) ;
    for ( ll y = 1 , i = 0 ; i < n ; ++ i , y *= 10 ) cnt[ 0 ] -= y ;
    ++ cnt[ 0 ] ;
}
 
int main(  ) {
    scanf( "%lld%lld" , &a , &b ) ;
    cal( b ) ; 
    rep( i , 10 ) ans[ i ] = cnt[ i ] ;
    cal( a - 1 ) ;
    rep( i , 9 ) printf( "%lld " , ans[ i ] - cnt[ i ] ) ;
    printf( "%lld\n" , ans[ 9 ] - cnt[ 9 ] ) ;
    return 0 ;
}

你可能感兴趣的:(BZOJ-1833: [ZJOI2010]count 数字计数(数位统计))