184竞赛

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

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

给你网格图的行数 n 。

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

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

/**
 * @param {number} n
 * @return {number}
 */
// 找规律 下层的结果 = ABA类的个数 + ABC类的个数 = (3m+2n) + (2m+2n)
var numOfWays = function(n) {
    if(n===0){
        return 0
    } else if(n===1){
        return 12
    }
    let temp = 1000000007;
    let repeat = 6;
    let unrepeat = 6;
    for(let i=2; i<=n; i++){
        let newrep =   repeat *3 % temp + unrepeat * 2 % temp;
        let newunrep = repeat *2 % temp + unrepeat *2 % temp;
        repeat = newrep;
        unrepeat = newunrep;
    }
    return (repeat + unrepeat)%temp
};

/**
 * @param {number} n
 * @return {number}
 */
// 我当时的想法是，排列组合么，但是第二行的第二列其实不是1。是有可能1，有可能2，缺少情况。
var numOfWays = function(n) {
 let arr  = new Array(n);
    for(let i=0; i<n; i++){
        arr[i] = new Array(3)
    }
    let row = arr.length;
    let clo = 3;
    for(let i=0; i<row; i++){
        for(let j=0; j<clo; j++){
            if(i==0 && j === 0){
                arr[i][j] = 3; 
            } else if(i==0 && j>0){
                arr[i][j] = 2;        
            } else if(i>0 && j===0){
                arr[i][j] = 2;        
            } else {
                arr[i][j] = 1;
            }
        }
    }
    
    let res = 1;
    for(let i=0; i<row; i++){
        for(let j=0; j<clo; j++){
            res*=arr[i][j];
            res%=Math.pow(10, 9)+7
        }
    }
    for(let i=0; i<row; i++){
        for(let j=0; j<clo; j++){
            if(i==0 && j === 0){
                arr[i][j] = 3; 
            } else if(i==0 && j>0){
                arr[i][j] = 2;        
            } else if(i>0 && j===0){
                arr[i][j] = 2;        
            } else {
                arr[i][j] = 1;
            }
        }
    }
    // arr[0][0] = 3;
//     for(let i=0; i<arr.length; i++){
//         for(let j=0; j<arr[0].length; j++){
//             if((arr[i-1] && arr[i-1][j] === undefined && arr[i] && arr[i][j-1] !== undefined) || (arr[i-1] && arr[i-1][j] !== undefined && arr[i] && arr[i][j-1] === undefined)){
//                 arr[i][j] = 2
//             } else if(arr[i-1] && arr[i-1][j] !== undefined && arr[i] && arr[i][j-1] !== undefined){
//                   arr[i][j] = 1    
//             } else if(arr[i-1] && arr[i-1][j] == undefined && arr[i] && arr[i][j-1] == undefined){
//                 arr[i][j] = 3  
//             }
            
//         }
//     }
    return res;
};

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

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

猜你喜欢