最近两场比赛,都涉及到此类问题!以前没注意过这种问题,都在状态转移方程推出后,止步于n太大!事后总结,线性递推可以用矩阵来表示,进而快速幂解决n太大的问题!
例1:牛客小白月赛7 J 方格填色
题意:n×m矩阵填黑白两色,行相同相邻两个不能同为白色,相邻两列不全为黑色
(n≤5,m≤1e18),问填色的方案数mod 1e9+7的值.
令黑色为1,白色为0,将一列的颜色01由下往上排布,即An-1An-2……A0
(Ai为1表示该列第i+1格子为黑色,0即为白色)
也就是2进制表示,00……0~11……1,每列有2^n种可能出现的值。(状态)
(每一列的长度最长为5,即状态数最多为32,2的5次方)
限制条件:
相邻两列不全为黑色—>第i列为111……1,第i+1列不能为状态111……1
相邻两列的同行格不同为白色—>第i列的j位为0,第i+1列的j位不为0
相邻状态关系结论:
前面状态为x,相邻状态为y。
x、y满足 x|y = 11……1(2进制),(x=y=11……1除外)
状态转移方程:
f[x][m] 定义:第m列涂色状态为x,前面1~m列染色满足限制条件的方案数
f [ x ] [ m ] = ∑ ( 0 ≤ y ≤ 2 n − 1 , x ∣ y = 2 n − 1 , x = y ≠ 2 n − 1 ) f [ y ] [ m − 1 ] , ( 0 ≤ x ≤ 2 n − 1 ) f[x][m] = \sum_{(0\leq y \leq 2^n-1,x|y=2^n-1,x=y\neq 2^n-1)}f[y][m-1],(0\leq x \leq 2^n-1) f[x][m]=(0≤y≤2n−1,x∣y=2n−1,x=y̸=2n−1)∑f[y][m−1],(0≤x≤2n−1)
举个例子: n=2时,每列有4种状态,枚举第m列的状态
1. 状 态 为 0 , f [ 0 ] [ m ] = f [ 3 ] [ m − 1 ] 1. 状态为0,f[0][m] = f[3][m-1] 1.状态为0,f[0][m]=f[3][m−1] ( 第 m 列 全 为 白 色 , 不 能 同 数 位 为 0 , 第 m − 1 列 全 为 黑 色 ) (第m列全为白色,不能同数位为0,第m-1列全为黑色) (第m列全为白色,不能同数位为0,第m−1列全为黑色)
2. 状 态 为 1 , f [ 1 ] [ m ] = f [ 2 ] [ m − 1 ] + f [ 3 ] [ m − 1 ] 2. 状态为1,f[1][m] = f[2][m-1]+f[3][m-1] 2.状态为1,f[1][m]=f[2][m−1]+f[3][m−1]
( 第 m 列 第 2 格 为 白 色 , 第 m − 1 列 第 2 格 必 须 为 黑 色 , 第 1 格 可 黑 可 白 ) (第m列第2格为白色,第m-1列第2格必须为黑色,第1格可黑可白) (第m列第2格为白色,第m−1列第2格必须为黑色,第1格可黑可白)
3. 状 态 为 2 , f [ 2 ] [ m ] = f [ 1 ] [ m − 1 ] + f [ 3 ] [ m − 1 ] 3. 状态为2,f[2][m] = f[1][m-1]+f[3][m-1] 3.状态为2,f[2][m]=f[1][m−1]+f[3][m−1]
( 第 m 列 第 1 格 为 白 色 , 第 m − 1 列 第 1 格 必 须 为 黑 色 , 第 2 格 可 黑 可 白 ) (第m列第1格为白色,第m-1列第1格必须为黑色,第2格可黑可白) (第m列第1格为白色,第m−1列第1格必须为黑色,第2格可黑可白)
4. 状 态 为 3 , f [ 3 ] [ m ] = f [ 0 ] [ m − 1 ] + f [ 1 ] [ m − 1 ] + f [ 2 ] [ m − 1 ] 4. 状态为3,f[3][m] = f[0][m-1]+f[1][m-1]+f[2][m-1] 4.状态为3,f[3][m]=f[0][m−1]+f[1][m−1]+f[2][m−1]
( 第 m 列 全 为 黑 色 , 即 只 要 第 m − 1 列 不 全 为 黑 色 即 可 ) (第m列全为黑色,即只要第m-1列不全为黑色即可) (第m列全为黑色,即只要第m−1列不全为黑色即可)
整理如下:
{ f [ 0 ] [ m ] = f [ 3 ] [ m − 1 ] f [ 1 ] [ m ] = f [ 2 ] [ m − 1 ] + f [ 3 ] [ m − 1 ] f [ 2 ] [ m ] = f [ 1 ] [ m − 1 ] + f [ 3 ] [ m − 1 ] f [ 3 ] [ m ] = f [ 0 ] [ m − 1 ] + f [ 1 ] [ m − 1 ] + f [ 2 ] [ m − 1 ] \begin{cases} f[0][m] &= f[3][m-1]\\ f[1][m] &= f[2][m-1]+f[3][m-1]\\ f[2][m] &= f[1][m-1]+f[3][m-1]\\ f[3][m] &= f[0][m-1]+f[1][m-1]+f[2][m-1] \end{cases} ⎩⎪⎪⎪⎨⎪⎪⎪⎧f[0][m]f[1][m]f[2][m]f[3][m]=f[3][m−1]=f[2][m−1]+f[3][m−1]=f[1][m−1]+f[3][m−1]=f[0][m−1]+f[1][m−1]+f[2][m−1]
当m不大时,可以去递推,但m超大时,必须另外想方法,因为是线性递推,而且递推状态不多,可以矩阵表示,考虑矩阵快速幂
[ f [ 0 ] [ m ] f [ 1 ] [ m ] f [ 2 ] [ m ] f [ 3 ] [ m ] ] = [ 0 0 0 1 0 0 1 1 0 1 0 1 1 1 1 0 ] ∗ [ f [ 0 ] [ m − 1 ] f [ 1 ] [ m − 1 ] f [ 2 ] [ m − 1 ] f [ 3 ] [ m − 1 ] ] \left[ \begin{matrix} f[0][m] \\ \\ f[1][m]\\ \\ f[2][m]\\ \\ f[3][m]\\ \end{matrix} \right] = \left[ \begin{matrix} 0&0&0&1\\ \\ 0&0&1&1\\ \\ 0&1&0&1\\ \\ 1&1&1&0\\ \end{matrix}\right] * \left[ \begin{matrix} f[0][m-1]\\ \\ f[1][m-1]\\ \\ f[2][m-1]\\ \\ f[3][m-1]\\ \end{matrix} \right] ⎣⎢⎢⎢⎢⎢⎢⎢⎢⎡f[0][m]f[1][m]f[2][m]f[3][m]⎦⎥⎥⎥⎥⎥⎥⎥⎥⎤=⎣⎢⎢⎢⎢⎢⎢⎢⎢⎡0001001101011110⎦⎥⎥⎥⎥⎥⎥⎥⎥⎤∗