Leetcode-552 Student Attendance Record II(学生出勤记录 II）

#define maxn 1000000
#define _for(i,a,b) for(int i = (a);i < (b);i ++)
#define pb push_back
#define MOD 1000000007

class Solution
{
    public:
        int checkRecord(int n)
        {
            long long int dp[2][2][3] {0};
            
            dp[0][0][0] = dp[0][0][1] = dp[0][1][0] = 1;
            
            _for(i,2,n+1)
            {
                if(!(i&0x1))
                {
                    dp[1][0][0] = (dp[0][0][0]+dp[0][0][1]+dp[0][0][2])%MOD;
                    dp[1][0][1] = dp[0][0][0];
                    dp[1][0][2] = dp[0][0][1];
                    dp[1][1][0] = ((dp[0][0][0]+dp[0][0][1])%MOD+(dp[0][0][2]+dp[0][1][0])%MOD+(dp[0][1][1]+dp[0][1][2])%MOD)%MOD;
                    dp[1][1][1] = dp[0][1][0];
                    dp[1][1][2] = dp[0][1][1];
                }
                else
                {
                    dp[0][0][0] = (dp[1][0][0]+dp[1][0][1]+dp[1][0][2])%MOD;
                    dp[0][0][1] = dp[1][0][0];
                    dp[0][0][2] = dp[1][0][1];
                    dp[0][1][0] = ((dp[1][0][0]+dp[1][0][1])%MOD+(dp[1][0][2]+dp[1][1][0])%MOD+(dp[1][1][1]+dp[1][1][2])%MOD)%MOD;
                    dp[0][1][1] = dp[1][1][0];
                    dp[0][1][2] = dp[1][1][1];
                }
            }
            if(!(n&0x1))
                n = 1;
            else
                n = 0;
            int rnt = (dp[n][0][0]+dp[n][0][1]+dp[n][0][2]
            +dp[n][1][0]+dp[n][1][1]+dp[n][1][2])%MOD;
            
            return rnt;
        }
};