LightOJ 1021 Painful Bases 【狀壓DP+數位DP】

題目鏈接

題意

求由一些B進制的數的全排列中能被K整除的數的個數

分析

題中B最高達到16，直接枚舉排列顯然不可能。考慮數位DP，但同時取得每個數要不同，所以需要記錄用過哪些數，因此要用到狀壓DP

狀態

dp[S][r]↔用了數集{S}中的數後除以K餘數爲r的數的個數

轉移方程

dp[S][r]=∑a∈{S}dp[S−{a}][r′]

其中 (r′+a×Bt−1%k)%k=r (t爲S中元素個數）

另外注意base的冪要預處理，不然會T

AC代碼

//LightOJ 1021 Painful Bases
//AC 2016-08-04 15:02:54
//DP
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cctype>
#include <cstdlib>
#include <cstring>
#include <vector>
#include <set>
#include <string>
#include <map>
#include <queue>
#include <deque>
#include <list>
#include <sstream>
#include <stack>
using namespace std;

#define cls(x) memset(x,0,sizeof x)
#define inf(x) memset(x,0x3f,sizeof x)
#define neg(x) memset(x,-1,sizeof x)
#define ninf(x) memset(x,0xc0,sizeof x)
#define st0(x) memset(x,false,sizeof x)
#define st1(x) memset(x,true,sizeof x)
#define INF 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define input(x) scanf("%d",&(x))
#define bug cout<<"here"<<endl;
//#define debug

unsigned long long dp[(1<<17)][30];
char org[20];
int num[20];
int T,B,K;

int Pow(int x,int y)
{
    int res=1;
    for(int i=0;i<y;++i)
        res=(res%K*x%K)%K;
    return res%K;
}
int pows[20];

int main()
{
    #ifdef debug
        freopen("E:\\Documents\\code\\input.txt","r",stdin);
        freopen("E:\\Documents\\code\\output.txt","w",stdout);
    #endif
    input(T);
    for(int kase=1;kase<=T;++kase)
    {
        input(B);input(K);
        scanf("%s",org);
        int len=strlen(org);
        for(int i=0;i<len;++i)
        {
            if(isdigit(org[i]))
                num[i]=org[i]-'0';
            else
                num[i]=org[i]-'A'+10;
        }
        cls(dp);
        dp[0][0]=1;
        for(int i=0;i<=len;++i)
            pows[i]=Pow(B,i);
        for(int i=0;i<(1<<len)-1;++i)
        {
            int t=0;
            for(int j=0;j<len;++j)
                if((i>>j)&1) ++t;
            for(int j=0;j<len;++j)
            {
                if((i>>j)&1) continue;
                for(int r=0;r<K;++r)
                {
                    int r1=((r+(num[j]%K)*pows[t])%K)%K;
                    dp[i|(1<<j)][r1]+=dp[i][r];
                }
            }
        }
        printf("Case %d: ",kase);
        cout<<dp[(1<<len)-1][0]<<endl;
    }
    return 0;
}