leetcode 5383. 给 N x 3 网格图涂色的方案数

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

给你网格图的行数 n 。

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

示例 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
 

 

题解：因为只有三种颜色和 列数只有3 所以把所有的对应情况列出来就好

a类 为有两个相同颜色的 如 xyx

b类 为三个颜色 不同 如 xyz

可以得到 a类的下一层 可以是 3种a类和2种b类

可以得到 b类的下一层 可以是 2种a类和2种b类

#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define ff(i,a,b) for(int i = a; i <= b; i++)
#define f(i,a,b) for(int i = a; i < b; i++)
typedef pair<int,int> P;
#define ll long long
const ll mod  = 1e9 + 7;
class Solution {
public:
    int numOfWays(int n) {
        ll a = 6,b = 6;
        ff(i,2,n)
        {
            ll c = 3 * a + 2 * b;
            ll d = 2 * a + 2 * b;
            c %= mod; d %= mod;
            a = c, b = d;
        }
        return (a + b) % mod;

    }
};