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

leetcode 5383. 给 N x 3 网格图涂色的方案数
来源:力扣(LeetCode)


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

给你网格图的行数 n 。

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

示例 1:

输入:n = 1
输出:12
解释:总共有 12 种可行的方法:
leetcode 5383. 给 N x 3 网格图涂色的方案数_第1张图片

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


思路分析

leetcode 5383. 给 N x 3 网格图涂色的方案数_第2张图片


代码解析
class Solution {
public:
    int numOfWays(int n) {
        int Max_num = 1e9 + 7;
        long int k = 6, m = 6,result = 12,temp;//第一列,第二列,结果总数,临时k值存储
        for(int i = 2;i <= n;++i){
            result = ( ((k * 5 ) % Max_num) + ((m * 4 ) % Max_num) )% Max_num;//计算总数
            temp = k;
            k = (((k * 3) % Max_num) + ((m * 2) % Max_num)) % Max_num;//更新k
            m = (((temp * 2) % Max_num) + ((m * 2) % Max_num)) % Max_num;//更新m
        }
        return result;
    }
};

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

你可能感兴趣的:(leetcode练习)