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

1.题目详情       leetcode 给N x 3网络图涂色的方案数

你有一个 n x 3 的网格图 grid ，你需要用 红，黄，绿 三种颜色之一给每一个格子上色，且确保相邻格子颜色不同（也就是有相同水平边或者垂直边的格子颜色不同）。给你网格图的行数 n 。请你返回给 grid 涂色的方案数。由于答案可能会非常大，请你返回答案对 10^9 + 7 取余的结果

2.解题思路

• 当层数为一时，可直接推算出共有12种方法，其中ABA型的涂法有6种，ABC型的涂法有6种。
• 然后继续写出第二层，和第三层的涂法。不难发现，ABA的涂法种，有3种ABA型，2种ABC型；ABC的涂法中，有2种ABA型，2种ABC型……依此类推求出最后的结果。

class Solution {
    public int numOfWays(int n) {
        if(n==1) return 12;
        long ABA=6;
        long ABC=6;
        int p=1000000007;
        for(int i=2;i<n+1;i++){
            long temp=ABA;
            ABA=((ABC*2)%p+(ABA*3)%p)%p;
            ABC=((ABC*2)%p+(temp*2)%p)%p;
        }
        return (int)(ABA+ABC)%p;
    }
}