bzoj1027 状压dp

https://www.lydsy.com/JudgeOnline/problem.php?id=1072

题意给一个数字串s和正整数d, 统计s有多少种不同的排列能被d整除

试了一下发现暴力可过

因为s的长度只有10的缘故，我们考虑用dp[i][j]来记录i状态下余数为j的数量。

i表示的是这个数字串中已经加入的数字，直接从前往后递推即可。

本题第一个难点在于想到用状压dp去做，第二个难点在于因为其中有相同的数字的缘故，算完了之后的答案需要用排列组合去重。

原本我觉得这样的状态无法表示每个元素加入的先后次序，后来发现每个dp状态的与下一个状态相关的只有余数，与加入的i先后状态无关，所以我们只需要考虑这个数选了没有即可

#include <map>
#include <set>
#include <ctime>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <sstream>
#include <iostream>
#include <algorithm>
#include <functional>
using namespace std;
#define For(i, x, y) for(int i=x;i<=y;i++)  
#define _For(i, x, y) for(int i=x;i>=y;i--)
#define Mem(f, x) memset(f,x,sizeof(f))  
#define Sca(x) scanf("%d", &x)
#define Sca2(x,y) scanf("%d%d",&x,&y)
#define Scl(x) scanf("%lld",&x);  
#define Pri(x) printf("%d\n", x)
#define Prl(x) printf("%lld\n",x);  
#define CLR(u) for(int i=0;i<=N;i++)u[i].clear();
#define LL long long
#define ULL unsigned long long  
#define mp make_pair
#define PII pair<int,int>
#define PIL pair<int,long long>
#define PLL pair<long long,long long>
#define pb push_back
#define fi first
#define se second 
typedef vector<int> VI;
const double eps = 1e-9;
const int maxn = 110;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7; 
int N,M,tmp,K; 
inline int read()
{
   int now=0;register char c=getchar();
   for(;!isdigit(c);c=getchar());
   for(;isdigit(c);now=now*10+c-'0',c=getchar());
   return now;
}
char str[maxn];
int a[maxn];
int d;
int ans;
int num[maxn];
LL dp[1025][1005];
const int C[] = {1,1,2,6,24,120,720,5040,40320,362880,3628800,};
int main()
{
    int T; Sca(T);
    while(T--){
        Mem(num,0); ans = 0;
        scanf("%s%d",str,&d);
        N = strlen(str);
        for(int i = 0 ; i < N ; i ++){
            a[i] = str[i] - '0';
            num[a[i]]++;
        }
        Mem(dp,0);
        dp[0][0] = 1;
        for(int i = 1 ; i < (1 << N); i ++){
            for(int j = 0; j < N ; j++){
                if(i & (1 << j)){
                    for(int k = 0 ; k < d; k ++){
                        dp[i][(k * 10 + a[j]) % d] += dp[i ^ (1 << j)][k];
                    }
                }
            }
        }
        LL ans = dp[(1 << N) - 1][0];
        For(i,0,9){
            if(num[i]) ans /= C[num[i]];
        }
        Prl(ans);
    }
    #ifdef VSCode
    system("pause");
    #endif
    return 0;
}

猜你喜欢