[BZOJ3294][Cqoi2011]放棋子（DP+组合数学+容斥）

最新推荐文章于 2021-07-07 16:56:10 发布

xyz32768

最新推荐文章于 2021-07-07 16:56:10 发布

阅读量225

点赞数

分类专栏： BZOJ、UOJ、LOJ 文章标签： DP 容斥

本文链接：https://blog.csdn.net/xyz32768/article/details/79293488

版权

BZOJ、UOJ、LOJ 专栏收录该内容

341 篇文章 1 订阅

订阅专栏

注意到条件「不同颜色的棋子不能放在同一行」，可以得出，一种颜色的棋子放完之后，这些棋子会占领一定的行和列，这些行和列不能继续放棋子。
所以定义状态 $f[k][i][j]$ 表示到第 $k$ 种颜色，还剩下 $i$ 行 $j$ 列没被占领。
边界为 $f[0][n][m]=1$ 。
转移为：

f [k] [i] [j] = \sum i < h \leq n \sum j < l \leq m f [k - 1] [h] [l] \times C i h \times C j l \times g [k] [h - i] [l - j]

$f[k][i][j]=\sum_{i<h\leq n}\sum_{j<l\leq m}f[k-1][h][l]\times C_h^i\times C_l^j\times g[k][h-i][l-j]$

g[k][i][j] g [ k ] [ i ] [ j ] $g[k][i][j]$ 表示第

k k $k$ 种颜色恰好占领

i

$i$ 行

j j $j$ 列的方案数。
关于

g

$g$ ， ~~一开始我想出了一个 $250\times n^2\times m^2$ 的求法然后TLE，~~可以用组合数+容斥原理求得。
先不考虑「恰好」，则为

Ccntki×j C i × j c n t k $C_{i\times j}^{cnt_k}$ （

cntk c n t k $cnt_k$ 为第

k k $k$ 种颜色的棋子个数）
然后除去不合法的方案数，即

g [k] [i] [j] = C_{i \times j}^{c n t_{k}} - \sum_{h = 1}^{i} \sum_{l = 1}^{j} g [k] [h] [l] \times C_{i}^{h} \times C_{j}^{l} [h \neq i or l \neq j]

$g[k][i][j]=C_{i\times j}^{cnt_k}-\sum_{h=1}^i\sum_{l=1}^jg[k][h][l]\times C_i^h\times C_j^l[h\neq i\mbox{ or }l\neq j]$
最后结果就是

∑ni=0∑mj=0f[c][i][j] ∑ i = 0 n ∑ j = 0 m f [ c ] [ i ] [ j ] $\sum_{i=0}^n\sum_{j=0}^mf[c][i][j]$ （不一定要占领所有的格子）。
代码：

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
inline int read() {
    int res = 0; bool bo = 0; char c;
    while (((c = getchar()) < '0' || c > '9') && c != '-');
    if (c == '-') bo = 1; else res = c - 48;
    while ((c = getchar()) >= '0' && c <= '9')
        res = (res << 3) + (res << 1) + (c - 48);
    return bo ? ~res + 1 : res;
}
const int R = 2005, N = 33, E = 13, MX = 1e9 + 9;
int n, m, c, orz, h, cnt[E], C[R][R], g[E][N][N], f[E][N][N];
int main() {
    int i, j, k, h, l; n = read(); m = read(); c = read();
    for (i = 1; i <= c; i++) orz += (cnt[i] = read());
    for (i = 0; i <= 2000; i++) C[i][0] = 1;
    for (i = 1; i <= 2000; i++) for (j = 1; j <= i; j++)
        C[i][j] = (C[i - 1][j] + C[i - 1][j - 1]) % MX;
    for (k = 1; k <= c; k++) for (i = 1; i <= n; i++)
    for (j = 1; j <= m; j++) {
        g[k][i][j] = C[i * j][cnt[k]];
        for (h = 1; h <= i; h++) for (l = 1; l <= j; l++) if (h != i || l != j)
            g[k][i][j] = (g[k][i][j] - 1ll * g[k][h][l]
            * C[i][h] % MX * C[j][l] % MX + MX) % MX;
    }
    f[0][n][m] = 1; for (k = 1; k <= c; k++) for (i = 0; i <= n; i++)
    for (j = 0; j <= m; j++) for (h = i + 1; h <= n; h++)
    for (l = j + 1; l <= m; l++)
        f[k][i][j] = (f[k][i][j] + 1ll * f[k - 1][h][l]
        * g[k][h - i][l - j] % MX * C[h][i] % MX * C[l][j] % MX) % MX;
    int ans = 0; for (i = 0; i <= n; i++) for (j = 0; j <= m; j++)
        ans = (ans + f[c][i][j]) % MX; cout << ans << endl;
    return 0;
}