JZOJ 1667【AHOI2009】中国象棋——dp

题目：https://jzoj.net/senior/#main/show/1667

只注重0、1、2的列有多少个，不注重它们的位置，就能记录了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=,mod=;
int n,m,dp[][N][N][N],ans,jc[N],jcn[N];
int pw(int x,int k)
{
    int ret=;while(k){if(k&)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=;}return ret;
}
void init()
{
    jc[]=;
    for(int i=;i<=m;i++)jc[i]=(ll)jc[i-]*i%mod;/////m not n
    jcn[m]=pw(jc[m],mod-);
    for(int i=m-;i>=;i--)jcn[i]=(ll)jcn[i+]*(i+)%mod;
}
int C(int n,int m)
{
    return (ll)jc[n]*jcn[m]%mod*jcn[n-m]%mod;
}
int main()
{
    scanf("%d%d",&n,&m);dp[][m][][]=;
    init();
    for(int i=;i<=n;i++)
    {
        int d=(i&);
        for(int x=;x<=m;x++)
            for(int y=;y<=m;y++)
            {
                int z=m-x-y;
                dp[d][x][y][z]=dp[!d][x][y][z];
                if(y)(dp[d][x][y][z]+=(ll)dp[!d][x+][y-][z]*(x+)%mod)%=mod;//0->1
                if(z)(dp[d][x][y][z]+=(ll)dp[!d][x][y+][z-]*(y+)%mod)%=mod;//1->2
                if(y>)(dp[d][x][y][z]+=(ll)dp[!d][x+][y-][z]*C(x+,)%mod)%=mod;//0,0->1,1
                if(y&&z)(dp[d][x][y][z]+=(ll)dp[!d][x+][y][z-]*(x+)%mod*y%mod)%=mod;//0,1->1,2
                if(z>)(dp[d][x][y][z]+=(ll)dp[!d][x][y+][z-]*C(y+,)%mod)%=mod;//1,1->2,2
            }
    }
    int d=(n&);
    for(int x=;x<=m;x++) for(int y=;y<=m;y++) (ans+=dp[d][x][y][m-x-y])%=mod;
    printf("%d\n",ans);
    return ;
}