[BZOJ1072]-[SCOI2007]排列perm-状压DP

说在前面

おもしろい

---

题目

BZOJ1072传送门

题面　

给一个数字串s和正整数d, 统计s有多少种不同的排列能被d整除（可以有前导0）。例如123434有90种排列能被2整除，其中末位为2的有30种，末位为4的有60种。

输入输出格式

输入格式：
输入第一行是一个整数T，表示测试数据的个数，以下每行一组s和d，中间用空格隔开。s保证只包含数字0, 1, 2, 3, 4, 5, 6, 7, 8, 9.
s的长度不超过10，d不大于1000

输出格式：
每个数据仅一行，表示能被d整除的排列的个数。

---

解法

看到整除想余数
看到整除想余数
看到整除想余数
重要的事情说三遍
对选择的数字集合进行状压，定义dp[state][len][rest] 表示当前已经使用了s中state位置上的数字，已经选择了len位数字，且当前余数为rest
然后就可以愉快的转移啦，记得最后答案要除以相同数字个数的阶乘（因为相同的数字在dp里算的是全排列，而它们是一样的）

---

下面是自带大常数的代码

/**************************************************************
    Problem: 1072
    User: Izumihanako
    Language: C++
    Result: Accepted
    Time:3056 ms
    Memory:45040 kb
****************************************************************/

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;

char ss[15] ;
int T , mmod , dp[11][1<<10][1005] , len , pow10[10] = {1} ;
int num[10] , cnt[10] , fac[10] = {1} ;

void solve(){
    dp[0][0][0] = 1 ;
    int Fstate = ( 1 << len ) - 1 ;
    register int i , s , r , k , nr ;
    for( i = 0 ; i <= len - 1 ; i ++ ){
        for( s = 0 ; s <= Fstate ; s ++ ){
            for( r = 0 ; r < mmod ; r ++ ){
                if( !dp[i][s][r] ) continue ;
                for( k = 0 ; k < len ; k ++ ){
                    if( s & ( 1 << k ) ) continue ;
                    nr = ( r + pow10[i]%mmod * num[k] )%mmod ;
                    dp[i+1][s|(1<<k)][nr] += dp[i][s][r] ;
                }
            }
        }
    }
    int ans = dp[len][Fstate][0] ;
    for( int i = 0 ; i <= 9 ; i ++ )
        if( cnt[i] ) ans /= fac[ cnt[i] ] ;
    printf( "%d\n" , ans ) ;
    memset( dp , 0 , sizeof( dp ) ) ;
}

int main(){
    scanf( "%d" , &T ) ;
    for( int i = 1 ; i <= 9 ; i ++ )
        pow10[i] = pow10[i-1] * 10 ,
        fac[i] = fac[i-1] * i ;

    while( T -- ){
        scanf( "%s" , ss ) , len = strlen( ss ) ;
        memset( cnt , 0 , sizeof( cnt ) ) ;
        for( int i = 0 ; i < len ; i ++ )
            num[i] = ss[len-1-i] - '0' , cnt[ num[i] ] ++ ;
        scanf( "%d" , &mmod ) ;
        solve() ;
    }
}

哦对，这道题me为了冲榜，疯狂的压了压常数，用队列优化了dp
然并卵，还是如此的慢= =

/**************************************************************
    Problem: 1072
    User: Izumihanako
    Language: C++
    Result: Accepted
    Time:1112 ms
    Memory:120328 kb
****************************************************************/

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;

char ss[15] ;
int T , mmod , len , pow10[10] = {1} ;
int num[10] , cnt[10] , dp[11][1<<10][1001] , fac[10] = {1} , fr , ba ;
struct Data{
    short i , s , r ;
    Data(){} ;
    Data( const int &i_ , const int &s_ , const int &r_ ):
        i(i_) , s(s_) , r(r_){} ;
}que[11000005] ;

bool inque[11][1<<10][1001] ;
void solve(){
    fr = 1 , ba = 0 ;
    int Fstate = ( 1 << len ) - 1 , fg = 0 ;
    dp[0][0][0] = 1 ;
    que[++ba] = Data( 0 , 0 , 0 ) ;
    register int i , s , r , ans ;
    while( fr - 1 != ba ){
        i = que[fr].i , s = que[fr].s , r = que[fr].r ;
        if( fr == 10000000 ) fr = 1 ;
        else fr ++ ;
        inque[i][s][r] = false ;
        ans = dp[i][s][r] ;
        dp[i][s][r] = 0 ;
        if( i == len && s == Fstate && r == 0 ){
            fg = 1 ;
            break ;
        }
        for( int k = 0 ; k < len ; k ++ ){
            if( s & ( 1 << k ) ) continue ;
            int nr = ( r + pow10[i]%mmod * num[k] )%mmod ;
            dp[i+1][s|(1<<k)][nr] += ans ;
            if( !inque[i+1][s|(1<<k)][nr] ){
                if( ba == 10000000 ) ba = 1 ;
                else ba ++ ;
                que[ba] = Data( i+1 , s|(1<<k) , nr ) ;
                inque[i+1][s|(1<<k)][nr] = true ;
            }
        }
    }
    while( fr - 1 != ba ){
        Data u = que[fr] ;
        if( fr == 10000000 ) fr = 1 ;
        else fr ++ ;
        inque[u.i][u.s][u.r] = false ;
        dp[u.i][u.s][u.r] = 0 ;
    }

    for( int i = 0 ; i <= 9 ; i ++ )
        if( cnt[i] ) ans /= fac[ cnt[i] ] ;
    printf( "%d\n" , fg ? ans : 0 ) ;
}

int main(){
    scanf( "%d" , &T ) ;
    for( int i = 1 ; i <= 9 ; i ++ )
        pow10[i] = pow10[i-1] * 10 ,
        fac[i] = fac[i-1] * i ;

    while( T -- ){
        scanf( "%s" , ss ) , len = strlen( ss ) ;
        memset( cnt , 0 , sizeof( cnt ) ) ;
        for( int i = 0 ; i < len ; i ++ )
            num[i] = ss[len-1-i] - '0' , cnt[ num[i] ] ++ ;
        scanf( "%d" , &mmod ) ;
        solve() ;
    }
}