//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;
}
};
leetcode1411+Number of Ways to Paint N × 3 Grid+dp找规律+还可矩阵快速幂运加速
最新推荐文章于 2024-10-04 14:10:29 发布