BZOJ P1072 排列perm【状态压缩】

题目分析:

最开始做这道题的时候第一反应就是直接用全排列扫一遍,假设最长的数串长度为最长,也就是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;
}

猜你喜欢

转载自blog.csdn.net/yanzhenhuai/article/details/80461148