题目描述
有一个\(3\times n\)的网格,一些格子里已经有棋子了,一些格子里还没有。
每次你可以选择往一个没有棋子的格子里放一个棋子,但要满足这个格子上下两个格子都有棋子或左右两个格子都有棋子。
你的任务是把这个网格填满。问你有几种填法。
\(n\leq 2000\)
题解
先判无解。
如果四个角没有棋子或在第\(1/3\)行有两个相邻的空格就无解。
然后DP。
可以对于每个连通块分开DP,然后把结果合并。
可以看出一个连通块只可能通过第\(2\)行相邻。
设\(f_{i,j,k}\)为前面\(i\)行,\((2,i)\)这个格子在前面所有空格中是第\(j\)个放的,\((2,i+1)\)是否需要在\((2,i)\)之前放 的方案数。
转移:枚举\((2,i+1)\)是在什么时候放的。
设\(c\)为第\(i+1\)列两边的空格数。
\(f_{i,j,1}\rightarrow f_{i+1,l,0}(l\leq j)\),上下都要先放:\(A(l-1,c)\)
\(f_{i,j,0}\rightarrow f_{i+1,l,1}(l>j)\),上下有一个后放:\(c(l-1)A(cnt-l,c-1)\),两个都后放:\(A(cnt-l,2)\)
$f_{i,j,0}\rightarrow f_{i+1,l,0} \(,上下都要先放:\)A(l-1,c)$
其中\(A(n,m)\)为排列数。
可以用前缀和优化DP。
还要考虑\((2,i)\)不是空格但\((1,i),(3,i)\)是空格的情况。
时间复杂度:\(O(n^2)\)
代码
$f_{i,j,1}\rightarrow f_{i+1,l,0}(l\leq j)$,上下要先放:$A(l-1,c)$
$f_{i,j,0}\rightarrow f_{i+1,l,1}(l>j)$,上下至少有一个没放:$c(l-1)A(cnt-l,c-1)$
$f_{i,j,0}\rightarrow f_{i+1,l,0}$,上下先放:$A(l-1,c)$