BZOJ P1072 排列perm【状态压缩】

最新推荐文章于 2021-08-22 14:05:19 发布

原创最新推荐文章于 2021-08-22 14:05:19 发布 · 448 阅读

1 ·

CC 4.0 BY-SA版权

OI/ACM之动态规划与递推专栏收录该内容

122 篇文章

订阅专栏

部署运行你感兴趣的模型镜像

题目分析：

最开始做这道题的时候第一反应就是直接用全排列扫一遍，假设最长的数串长度为最长，也就是 $10$ ，那么时间复杂度最坏是 $（10!*15）=（54432000）$ ，想着这样的时间复杂度在 $BZOJ$ 的三元评测机上交说不准就超时了，就没有这样写全排列（不过据说这道题确实可以 $next\ permutation$ 扫一遍，中间用 $set$ 判重，不过我不会 $set$ ，改天学一下）。

所以最后我还是用了状态压缩来做，这道题的数据范围（数串的最长长度为 $10$ ）已经提示了用状态压缩。

设立一个状态（突然不知道怎么用语言描述这个状态） $DP(I,J)，I$ 表示当前枚举的状态， $J$ 表示状态I对应的数字 $%Mod$ 的数值，这样设立状态后很容易得出下面的状态转移方程：
这里写图片描述

其中判断语句的含义是在 $I$ 状态中， $K$ 号没有选择出来。

状态转移结束后，由于数串中可能存在重复的数字（样例已经给出来了），这个时候我们就会有许多重复的计算。这个问题很好解决，我们根据排列的知识将最后的 $Ans/=Cnt[I]$ （ $Cnt[I]$ 记录数字I在数串中出现的次数）就可以了。

参考代码：

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define DB double
#define SG string
#define LL long long
#define DP(A,B) DP[A][B]
#define Fp(A,B,C,D) for(A=B;A<=C;A+=D)
#define Fm(A,B,C,D) for(A=B;A>=C;A-=D)
#define Clear(A) memset(A,0,sizeof(A))
using namespace std;
const LL Mod=1e9+7;
const LL Max=2e3+5;
const LL Inf=1e18;
LL T,M,Num[Max],Cnt[Max],DP[Max][Max];
inline LL Read(){
    LL X=0;char CH=getchar();bool F=0;
    while(CH>'9'||CH<'0'){if(CH=='-')F=1;CH=getchar();}
    while(CH>='0'&&CH<='9'){X=(X<<1)+(X<<3)+CH-'0';CH=getchar();}
    return F?-X:X;
}
inline void Write(LL X){
    if(X<0)X=-X,putchar('-');
    if(X>9)Write(X/10);
    putchar(X%10+48);
}
int main(){
    LL I,J,K,L;
    T=Read();
    while(T--){
        Clear(Cnt);Clear(DP);
        char CH[15];scanf("%s",CH+1);
        LL Length=strlen(CH+1);M=Read();
        Fp(I,1,Length,1){
            Num[I]=CH[I]-'0';
            Cnt[Num[I]]++;
        }DP(0,0)=1;
        K=(1<<Length)-1;
        Fp(I,0,K,1){
            Fp(J,0,M-1,1){
                Fp(L,1,Length,1){
                    if((I&(1<<(L-1)))==0){
                        DP(I|(1<<(L-1)),((J<<1)+(J<<3)+Num[L])%M)+=DP(I,J);
                    }
                }
            }
        }LL Ans=DP(K,0);
        Fp(I,0,9,1){
            Fp(J,1,Cnt[I],1){
                Ans/=J;
            }
        }Write(Ans);putchar('\n');
    }
    return 0;
}