可以用字符串表示一个学生的出勤记录,其中的每个字符用来标记当天的出勤情况(缺勤、迟到、到场)。记录中只含下面三种字符:
‘A’:Absent,缺勤
‘L’:Late,迟到
‘P’:Present,到场
如果学生能够 同时 满足下面两个条件,则可以获得出勤奖励:
按 总出勤 计,学生缺勤(‘A’)严格 少于两天。
学生 不会 存在 连续 3 天或 连续 3 天以上的迟到(‘L’)记录。
给你一个整数 n ,表示出勤记录的长度(次数)。请你返回记录长度为 n 时,可能获得出勤奖励的记录情况 数量 。答案可能很大,所以返回对 109 + 7 取余 的结果。
示例 1:
输入:n = 2
输出:8
解释:
有 8 种长度为 2 的记录将被视为可奖励:
“PP” , “AP”, “PA”, “LP”, “PL”, “AL”, “LA”, “LL”
只有"AA"不会被视为可奖励,因为缺勤次数为 2 次(需要少于 2 次)。
示例 2:
输入:n = 1
输出:3
示例 3:
输入:n = 10101
输出:183236316
提示:
1 <= n <= 105
思路:状态转移方程应该记录3个属性,i是多少个字符了,j 是A的数量,k是连续L的数量。状态转移是,我们可以枚举下一个字符填什么,这样的话容易分析。如果从i j k 状态过来了,可以填 A L P,但不是每一种填法都是合法的,因此需要判断是否合法,如果A的个数(j)为0,那么可以填A,如果连续L的数量小于3,可以填L。否则就是正常转移。
const int mod = 1e9 + 7, N = 100010;
int f[N][2][3];
class Solution {
public:
int checkRecord(int n) {
memset(f, 0, sizeof(f));
f[0][0][0] = 1;
for(int i = 0; i < n; i ++ )
for(int j = 0; j < 2; j ++ )
for(int k = 0; k < 3; k ++ ) {
if(!j) f[i + 1][j + 1][0] = (f[i + 1][j + 1][0] + f[i][j][k]) % mod;
if(k + 1 <= 2) f[i + 1][j][k + 1] = (f[i + 1][j][k + 1] + f[i][j][k]) % mod;
f[i + 1][j][0] = (f[i + 1][j][0] + f[i][j][k]) % mod;
}
int res = 0;
for(int j = 0; j < 2; j ++ )
for(int k = 0; k < 3; k ++ )
res = (res + f[n][j][k]) % mod;
return res;
}
};