LeetCode_1411给 N x 3 网格图涂色的方案数

你有一个 n x 3 的网格图 grid ，你需要用 红，黄，绿 三种颜色之一给每一个格子上色，且确保相邻格子颜色不同（也就是有相同水平边或者垂直边的格子颜色不同）。

给你网格图的行数 n 。

请你返回给 grid 涂色的方案数。由于答案可能会非常大，请你返回答案对 10^9 + 7 取余的结果。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/number-of-ways-to-paint-n-x-3-grid
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
示例 1：

输入：n = 1
输出：12
解释：总共有 12 种可行的方法：

示例 2：

输入：n = 2
输出：54

示例 3：

输入：n = 3
输出：246

示例 4：

输入：n = 7
输出：106494

示例 5：

输入：n = 5000
输出：30228214

提示：

n == grid.length
grid[i].length == 3
1 <= n <= 5000

方案一：
有题可知每层出现的情况只有aba、abc类的数据。

1. 第一层：6个aba类+6个abc类
2. 本层的每个aba类会产生下层 3个aba+2个abc
   本层的每个abc类会产生下层 2个aba+2个abc
3. 把每层的aba、abc数量分别求和 推出下层aba、abc的数量

一、

class Solution {
   public int numOfWays(int n) {
	   int divisor = 1000000007;
		long aba = 6;
		long abc = 6;
		for (int i = 2; i <= n; i++) {
			int tempABA = (int)((3 * aba + 2 * abc) % divisor);
			abc = (int)((2 * aba + 2 * abc) % divisor);
			aba = tempABA;
		}
		return (int)((aba + abc) % divisor);
	}
}

结果：

50 / 50 个通过测试用例
状态：通过
执行用时: 3 ms
内存消耗: 35.2 MB

二、
更易于理解

class Solution {
   public int numOfWays(int n) {
		if (n == 1) {
			return 12;
		}
		return (int)(6 * nameABA(n - 1, 1) + 6 * nameABC(n - 1, 1))%(1000000007);
	}

	private long nameABA(int n, int index) {
		if (index  >= n) {
			return 5;
		}
		return 3 * nameABA(n, index + 1) + 2 * nameABC(n, index + 1);
	}

	private long nameABC(int n, int index) {
		if (index  >= n) {
			return 4;
		}
		return 2 * nameABA(n, index + 1) + 2 * nameABC(n, index + 1);
	}
}

结果：

4 / 50 个通过测试用例
超时

方案二：

结果：

超时