洛谷P1446 [HNOI2008]Cards

置换群+dp

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>
#define MAXN 65
#define ll long long
#define pb push_back
#define ft first
#define sc second
#define mp make_pair
using namespace std;
int Inv(int a,int p){
    int b=p-2;
    int ret=1;
    while(b){
        if(b&1){
            ret=1LL*ret*a%p;
        }    
        a=1LL*a*a%p;
        b>>=1;
    }
    return ret;
}
int sr,sb,sg,n,m,MOD;
int a[MAXN][MAXN];
int f[25][25][25];
int b[MAXN],d[MAXN];
int calc(int x){
    memset(b,0,sizeof(b));
    memset(d,0,sizeof(d));
    int cnt=0;
    for(int i=1;i<=n;i++){
        if(!b[i]){
            cnt++;
            for(int j=i;!b[j];j=a[x][j]){
                d[cnt]++;
                b[j]=1;
            }
        }
    }
    memset(f,0,sizeof(f));
    f[0][0][0]=1;
    for(int h=1;h<=cnt;h++){
        for(int i=sr;i>=0;i--){
            for(int j=sb;j>=0;j--){
                for(int k=sg;k>=0;k--){
                    if(i>=d[h])(f[i][j][k]+=f[i-d[h]][j][k])%=MOD;
                    if(j>=d[h])(f[i][j][k]+=f[i][j-d[h]][k])%=MOD;
                    if(k>=d[h])(f[i][j][k]+=f[i][j][k-d[h]])%=MOD;
                }
            }
        }    
    }
    return f[sr][sb][sg];
}
int main()
{
//    freopen("data.in","r",stdin);
    scanf("%d%d%d%d%d",&sr,&sb,&sg,&m,&MOD);
    n=sr+sb+sg;
    int t;
    for(int i=1;i<=m;i++){
        for(int j=1;j<=n;j++){
            scanf("%d",&t);
            a[i][t]=j;
        }
    }
    m++;
    for(int i=1;i<=n;i++)a[m][i]=i;
    int ans=0;
    for(int i=1;i<=m;i++){
        ans+=calc(i);
        ans%=MOD;
    }
    ans=1LL*ans*Inv(m,MOD)%MOD;
    printf("%d
",ans);
    return 0;
}