leetcode-给Nx3网格图涂色的方案数_leetcode格子染色方案次数-CSDN博客

本文链接：https://blog.csdn.net/JR_Chan/article/details/105638876

题目是LeetCode第184场周赛的第四题，链接：给 N x 3 网格图涂色的方案数。具体描述可见原题。

这道题可以用观察+数学递推公式直接给出最优解。题目已经给出了一行（有3个网格）的所有可能12种涂色方案，那么对于下一行，其可能的涂色方案怎么计算呢。我们需要将这12种涂色方案先给分成两类，即ABC（用了所有三种颜色，如红黄蓝）和ABA（用了两种颜色，如红黄红），分别看这两类的下一行有什么可能性。首先是ABC，下一行可以是BCA/CAB/BAB/BCB，一共两种ABC类型的和两种ABA类型的；然后是ABA，下一行可以是BAB/BCB/CAC/BAC/CAB，一共两种ABC类型的和三种ABA类型的，所以下一行的ABC类型的数目 $ABC_{i+1}=2\times ABC_{i}+2\times ABA_{i}$ ，ABA类型的数目为 $ABA_{i+1}=2\times ABC_{i}+3\times ABA_{i}$ 。根据以上递推公式即可得出结果。时间复杂度为 $O (n)$ ，空间复杂度为 $O (1)$ 。

JAVA版代码如下：

class Solution {
    
    public int numOfWays(int n) {
        int ABC = 6, ABA = 6;
        int mod = 1_000_000_007;
        for (int i = 1; i < n; ++i) {
            int temp = ABC;
            ABC = (int)((ABC * 2L + ABA * 2L) % mod);
            ABA = (int)((temp * 2L + ABA * 3L) % mod);
        }
        return (int)(ABC + ABA) % mod;
    }
}

提交结果如下:

Python版代码如下：

class Solution:
    def numOfWays(self, n: int) -> int:
        ABC, ABA = 6, 6
        mod = 1000000007
        for _ in range(n - 1):
            ABC, ABA = 2 * ABC + 2 * ABA, 2 * ABC + 3 * ABA
        return (ABC + ABA) % mod

提交结果如下: