P3158 [CQOI2011] 放棋子

\(\text{solution}\)

首先这道题的状态是非常妙的。

设 \(f_{i, j, k}\) 表示任选 \(i\) 行 \(j\) 列，放 \(k\) 种颜色的方案数，但是明显不太好转移，我们可以设 \(g_{i, j, k}\) 表示任选 \(i\) 行 \(j\) 列，放 \(k\) 种相同颜色的方案数，然后用组合数乱草就可以了。