Light OJ 1021 Painful Bases (状压DP)

题意：给你一个base进制的数，求这个数各个位的所有排列形成的数能被k整除的个数。

解析：

一上来往数位DP上想去了，一看数据发觉不对，才反映到是状压DP。

将各个位的使用用二进制表示为状态S。

dp[S][t]为状态S下所有的数mod k = t的个数

则递推方程为 dp[S|(1<<i)][j+a[i]*base^x] += dp[S][j]。

[code]:

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;
typedef long long LL;

char s[20];
int n,base,m,a[20];
LL dp[1<<16][20];

void process(char *p){
    n = 0;
    while(*p != '\0'){
        if(*p<='9'&&*p>='0') a[n++] = *p-'0';
        else a[n++] = *p-'A'+10;
        p++;
    }
}
int mod_pow(int a,int b,int mod){
    int res = 1;
    while(b){
        if(b&1) res = (res*a)%mod;
        a = (a*a)%mod;
        b >>= 1;
    }
    return res;
}
LL sol(){
    int i,j,S,Ed,cnt,S_,tmp;
    Ed = 1<<n;
    memset(dp,0,sizeof(dp));
    dp[0][0] = 1;
    for(S = 0;S < Ed;S++){
        cnt = 0;
        for(i = 0;i < n;i++) cnt += S>>i&1;
        for(i = 0;i < n;i++){
            if(!(S>>i&1)){
                S_ = S|(1<<i);
                tmp = a[i]*mod_pow(base,cnt,m)%m;
                for(j = 0;j < m;j++){
                    dp[S_][(j+tmp)%m] += dp[S][j];
                }
            }
        }
    }
    return dp[Ed-1][0];
}
int main(){
    int i,j,cas,T;
    scanf("%d",&cas);
    for(T = 1;T <= cas;T++){
        scanf("%d%d",&base,&m);
        scanf("%s",s);
        process(s);
//puts("debug1");
        printf("Case %d: %lld\n",T,sol());
    }
    return 0;
}