552 Student Attendance Record II 学生出勤记录 II

给定一个正整数 n,返回长度为 n 的所有可被视为可奖励的出勤记录的数量。 答案可能非常大,你只需返回结果mod 109 + 7的值。
学生出勤记录是只包含以下三个字符的字符串:
    1.'A' : Absent,缺勤
    2.'L' : Late,迟到
    3.'P' : Present,到场
如果记录不包含多于一个'A'(缺勤)或超过两个连续的'L'(迟到),则该记录被视为可奖励的。
示例:
输入: n = 2
输出: 8
解释:
有8个长度为2的记录将被视为可奖励:
"PP" , "AP", "PA", "LP", "PL", "AL", "LA", "LL"
只有"AA"不会被视为可奖励,因为缺勤次数超过一次。
注意:n 的值不会超过100000。
详见:https://leetcode.com/problems/student-attendance-record-ii/description/

C++:

class Solution {
public:
    int checkRecord(int n) 
    {
        int M = 1000000007;
        int dp[n + 1][2][3] = {0};
        for (int j = 0; j < 2; ++j) 
        {
            for (int k = 0; k < 3; ++k) 
            {
                dp[0][j][k] = 1;
            }
        }
        for (int i = 1; i <= n; ++i)
        {
            for (int j = 0; j < 2; ++j) 
            {
                for (int k = 0; k < 3; ++k)
                {
                    int val = dp[i - 1][j][2];
                    if (j > 0)
                    {
                        val = (val + dp[i - 1][j - 1][2]) % M;
                    }
                    if (k > 0)
                    {
                        val = (val + dp[i - 1][j][k - 1]) % M;
                    }
                    dp[i][j][k] = val;
                }
            }
        }
        return dp[n][1][2];
    }
};

 参考:http://www.cnblogs.com/grandyang/p/6866756.html

posted on 2018-04-23 22:15  lina2014  阅读(269)  评论(0编辑  收藏  举报

导航