leetcode1411+Number of Ways to Paint N × 3 Grid+dp找规律+还可矩阵快速幂运加速

链接

//dp,找规律
//第一层，有 12 中情况，分为两大类：6 种 ABC 的情况，6 种 ABA 的情况
//当前层只与上一层和本层有关
//当上一层为 ABC类型时，当前层会产生 2 种 ABC 和 2 种 ABA
//当上一层为 ABA类型时，当前层会产生 2 种 ABC 和 3 种 ABA
//设 dp1[i] 表示第 i 层 ABC 的情况，dp2[i] 表示第 i 层 ABA 的情况
//则有 dp1[i] = dp1[i-1] * 2 + dp2[i-1] * 2，dp2[i] = dp1[i-1] * 2 + dp2[i-1] * 3
//最后dp1[n-1]+dp2[n-1]
class Solution {
public:
    int numOfWays(int n) {
        long long mod=1e9+7;
        vector<long long>dp1(n, 1);
        vector<long long>dp2(n,1);
        dp1[0]=6, dp2[0]=6;
        for(int i=1;i<n;i++){
            dp1[i]=(dp1[i-1]*2+dp2[i-1]*2)%mod;
            dp2[i]=(dp1[i-1]*2+dp2[i-1]*3)%mod;
        }
        return (dp1[n-1]+dp2[n-1])%mod;
    }
};