CodeForces 401 D.Roman and Numbers（状压DP）

Description

给出一正整数$n$，求满足以下三个条件的$x$的个数

1.$x$相当于是$n$的各位数字打乱

2.$x$无前导零

3.$x$被$m$整除

Input

两个整数$n,m(1\le n\le 10^{18}1,\le m\le 100)$

Output

输出满足条件的$x$个数

Sample Input

104 2

Sample Output

3

Solution

假设$n$有$res$位，那么以$res$个位表示$n$的每一位数字是否已经使用，考虑从$x$的低位往高位转移，$dp[i][j]$表示当前使用数字状态为$i$，模$m$为$j$的方案数，那么答案为$dp[2^{res}-1][0]$，由于$x$不能有前导零，故$dp$的初值只对非零数字为 1 <script type="math/tex" id="MathJax-Element-34">1</script>，之后就是对于一个状态，找到该状态还没使用的数字加进去然后转移即可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=(1<<18)+5;
ll n,dp[maxn][105];
int m,num[maxn],sum[maxn];
int main()
{
    while(~scanf("%I64d%d",&n,&m))
    {
        memset(num,0,sizeof(num));
        int N=0;
        while(n)N++,num[n%10]++,n/=10;
        sum[0]=0;
        for(int i=1;i<=10;i++)sum[i]=sum[i-1]+num[i-1]; 
        memset(dp,0,sizeof(dp));
        for(int i=1;i<10;i++)
            if(num[i])dp[1<<sum[i]][i%m]=1;
        N=1<<N;
        for(int i=0;i<N;i++)
            for(int j=0;j<m;j++)
                if(dp[i][j])
                    for(int k=0;k<=9;k++)
                        for(int l=sum[k];l<sum[k+1];l++)
                        {
                            if(i&(1<<l))continue;
                            dp[i^(1<<l)][(10*j+k)%m]+=dp[i][j];
                            break;
                        }
        printf("%I64d\n",dp[N-1][0]);
    }
    return 0;
}