P3158 [CQOI2011]放棋子（dp+组合数）

Posted 2022-11-10 kafuuchino

P3158 [CQOI2011]放棋子

放棋子的顺序和方案数无关，所以可以从按颜色递推

设$f[u][p][k]$为放到第$u$种颜色，所剩空间$p*k$的方案数

$g[u][i][j]$表示第$u$种颜色占据$i*j$空间的方案数，可以预处理

$g[u][i][j]=\binomi*jc[u]-\sum_p=1^i\sum_k=1^jg[u][p][k]*\binomii-p*\binomjj-k*[p<i||j<k]$

$f[u][p][k]=\sum_i=1^n-p\sum_j=1^m-kf[u-1][p+i][k+j]*\binomp+ii*\binomk+jj*g[u][i][j]$

复杂度$O(n^2m^2C)$

#include<iostream>
#include<cstdio>
#include<cstring>
#define ri register int
using namespace std;
const int P=1000000009;
int t,n,m,s,C[903][903],g[12][32][32],f[12][32][32],ans;
int main()
    scanf("%d%d%d",&n,&m,&t);
    for(ri i=0;i<=n*m;++i)
        C[i][0]=1;
        for(int j=1;j<=i;++j) C[i][j]=(C[i-1][j]+C[i-1][j-1])%P;
    f[0][n][m]=1;
    for(ri u=1;u<=t;++u)
        scanf("%d",&s);
        for(ri i=1;i<=n;++i)
            for(ri j=1;j<=m;++j) if(s<=i*j)
                g[u][i][j]=C[i*j][s];
                for(ri p=1;p<=i;++p)
                    for(ri k=1;k<=j;++k) if(p<i||k<j)
                        ri v=1ll*C[i][i-p]*C[j][j-k]%P;
                        (g[u][i][j]-=1ll*g[u][p][k]*v%P-P)%=P;
                    
            
        for(ri i=1;i<=n;++i)
            for(ri j=1;j<=m;++j)
                if(s<=i*j)
                    for(ri p=i;p<=n;++p)
                        for(ri k=j;k<=m;++k)
                            ri v=1ll*C[p][i]*C[k][j]%P*g[u][i][j]%P;
                            (f[u][p-i][k-j]+=1ll*f[u-1][p][k]*v%P)%=P;
                        
    
    for(ri i=0;i<=n;++i)
        for(ri j=0;j<=m;++j)
            (ans+=f[t][i][j])%=P;
    printf("%d",ans);
    return 0;