BZOJ 1072: [SCOI2007]排列perm 状压dp

1072: [SCOI2007]排列perm

Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 2538 Solved: 1583
[Submit][Status][Discuss]

Description

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

Input

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

Output

　　每个数据仅一行，表示能被d整除的排列的个数。

Sample Input

7

000 1

001 1

1234567890 1

123434 2

1234 7

12345 17

12345678 29

Sample Output

1

3

3628800

90

3

6

1398

HINT

在前三个例子中，排列分别有1, 3, 3628800种，它们都是1的倍数。

【限制】

100%的数据满足：s的长度不超过10, 1<=d<=1000, 1<=T<=15

Source

题解：定义dp[i][j]表示选择的状态为i，余数为j的方案数。则最后的答案就是dp[tot][0]里。
注意对于每一个重复的元素，若这个元素出现了x次，我们则重复计算了x!次，总共就重复计算了cnt[0]! * cnt[1]!*…*cnt[9]!所以最后要去重。

#include<bits/stdc++.h>
#define ll long long
using namespace std;

const int N = 11;
const int M = (1<<10);
const int D = 1000 + 2;

int t;
char s[N];
int n,a[N],d;
ll f[M][D],cnt[N],mpow[N];

#define ms(x,y) memset(x,y,sizeof(x))
void update(){
    ms(f,0);ms(cnt,0);
    for(int i=0;i<=9;++i) mpow[i]=1;
}

int main(){
    scanf("%d",&t);
    while(t--){
       update();
       scanf("%s%d",s,&d);n=strlen(s);
       for(int i=1;i<=n;++i) a[i]=s[i-1]-'0';
       int tot=(1<<n)-1;
       for(int i=1;i<=n;++i) {
          f[1<<i-1][a[i]%d]=1;
          ++cnt[a[i]],mpow[a[i]]=mpow[a[i]]*cnt[a[i]];
       }

       for(int i=0;i<=tot;++i){
         for(int k=1;k<=n;++k){
            if(i&(1<<k-1)) continue;
            for(int dd=0;dd<d;dd++){
              f[i|(1<<k-1)][(dd*10+a[k])%d]+=f[i][dd];
            }
         }
       }
       for(int i=0;i<=9;i++) f[tot][0]/=mpow[i];
       cout<<f[tot][0]<<endl;
    }
    return 0;
}