bzoj 4806: 炮dp

Posted 2020-12-28 lokiii

同1801
注意到一行只能放012个炮，我们只需要知道列的状态，不用状压行
所以设f[i][j][k]表示前i行有j列有1个炮，有k列有2个炮的方案数
然后分情况讨论转移就行了

#include<cstdio>
#include<iostream>
using namespace std;
const int N=105,mod=999983;
long long n,m,f[N][N][N],ans;
int main()
{
    scanf("%lld%lld",&n,&m);
    f[0][0][0]=1;
    for(int i=1;i<=n;i++)
        for(int j=0;j<=m;j++)
            for(int k=0;k<=m-j;k++)
            {
                f[i][j][k]=f[i-1][j][k];
                if(j>=1)
                    f[i][j][k]=(f[i][j][k]+(f[i-1][j-1][k]*(m-j+1ll-k))%mod)%mod;
                if(k>=1&&j+1<=m)
                    f[i][j][k]=(f[i][j][k]+(f[i-1][j+1][k-1]*(j+1ll))%mod)%mod;
                if(j>=2)
                    f[i][j][k]=(f[i][j][k]+(f[i-1][j-2][k]*((m-j+2ll-k)*(m-j+1ll-k)/2)%mod)%mod)%mod;
                if(k>=2&&j+2<=m)
                    f[i][j][k]=(f[i][j][k]+(f[i-1][j+2][k-2]*((long long)(j+2ll)*(long long)(j+1ll)/2)%mod)%mod)%mod;
                if(j>=1&&k>=1)
                    f[i][j][k]=(f[i][j][k]+(f[i-1][j][k-1]*(long long)j%mod*(m-j-k+1)%mod)%mod)%mod;
                if(i==n)
                    ans=(ans+f[i][j][k])%mod;
            }
    printf("%lld
",ans);
    return 0;
}