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