洛谷P3158 [CQOI2011]放棋子 组合数学+DP

题意：在一个m行n列的棋盘里放一些彩色的棋子，使得每个格子最多放一个棋子，且不同颜色的棋子不能在同一行或者同一列。有多少祌方法？

解法：这道题不会做，太菜了qwq。题解是看洛谷大佬的。

设C是组合数，f[i][j][k]:代表前k种棋子合法地恰好占领i行j列 

那么得到状态转移方程：f[i][j][k]=sigma f[ki][kj][k-1] * C[n-ki][i-ki] * C[m-kj][j-kj] * a[k]个棋子恰好占领i-ki行j-kj列的方案数。 这个式子的意思是我们枚举前k-1种棋子的占领情况是行占领ki行列占领kj列，那么第k种棋子就能占领i-ki行j-kj列，我们选出这i-ki/j-kj之后乘上通知颜色棋子a[k]个占领这i-ki/j-kj的方案数。

我们发现前面都都比较好算，唯独 a[k]个棋子恰好占领i-ki行j-kj列的方案数 这一项难算。

那么我们就考虑单独先预处理出这一项，设g[i][j][k]:代表k个同色棋子恰好占领了i行j列 ；

那么写出状态转移方程：g[i][j][k]=C[i*j][k] - sigma g[ki][kj][k] * C[i][i-ki] * C[j][j-kj] ；式子的意思是总的方案数减去不合法方案数，即同样是k个棋子却没有占满i行j列。

那么我们预处理出C数组和g数组，就可以获得AC了。

#include<bits/stdc++.h>
using namespace std;
const int N=100+10;
const int P=1e9+9;
typedef long long LL;
int n,m,c,a[N];
int C[N*10][N*10],f[N][N][N*10],g[N][N][N*10];
//g[i][j][k]:代表k个同色棋子恰好占领了i行j列 
//f[i][j][k]:代表前k种棋子合法地恰好占领i行j列 

void prework() {
    for (int i=0;i<=1000;i++)
        for (int j=0;j<=1000;j++)
            if (j==0 || i==j) C[i][j]=1;
            else C[i][j]=(C[i-1][j-1]+C[i-1][j])%P;
}

int main()
{
    cin>>n>>m>>c;
    int sum=0;
    for (int i=1;i<=c;i++) scanf("%d",&a[i]),sum+=a[i];
    prework();
    
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            for (int k=1;k<=sum;k++) {
                if (i*j<k) continue;
                g[i][j][k]=C[i*j][k];
                for (int ki=1;ki<=i;ki++)
                for (int kj=1;kj<=j;kj++)
                    if (ki!=i || kj!=j)    g[i][j][k]=(g[i][j][k]-(LL)g[ki][kj][k]*C[i][ki]%P*C[j][kj]%P)%P;
                g[i][j][k]=(g[i][j][k]%P+P)%P;
            }
    
    LL ans=0;
    f[0][0][0]=1;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            for (int k=1;k<=c;k++) {
                for (int ki=0;ki<=i;ki++)
                for (int kj=0;kj<=j;kj++)
                    if ((i-ki)*(j-kj)>=a[k])
                    f[i][j][k]=(f[i][j][k]+(LL)f[ki][kj][k-1]*C[n-ki][i-ki]%P*C[m-kj][j-kj]%P*g[i-ki][j-kj][a[k]]%P)%P;
                if (k==c) ans=(ans+f[i][j][k])%P;
            }
    cout<<ans<<endl;
    return 0;
}