leetcode 552 学生出勤记录II

题目描述

给定一个正整数 n，返回长度为 n 的所有可被视为可奖励的出勤记录的数量。 答案可能非常大，你只需返回结果mod 109 + 7的值。

学生出勤记录是只包含以下三个字符的字符串：

‘A’ : Absent，缺勤
‘L’ : Late，迟到
‘P’ : Present，到场
如果记录不包含多于一个’A’（缺勤）或超过两个连续的’L’（迟到），则该记录被视为可奖励的。

示例 1:

输入: n = 2
输出: 8 
解释：
有8个长度为2的记录将被视为可奖励：
"PP" , "AP", "PA", "LP", "PL", "AL", "LA", "LL"
只有"AA"不会被视为可奖励，因为缺勤次数超过一次。
注意：n 的值不会超过100000。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/student-attendance-record-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

状态定义
f(len,a_num,l_num)
表示len长度下的字符串中，A的个数为a_num且结尾中连续L的个数为l_num的字符串数量
限制条件：a_num<2;l_num<3

状态转移
当前状态后面还可以加什么字符(不能破坏限制条件哦)？
1.要是可以加A（满足条件a_num<1）
转移成：f(len+1,a_num+1,0)
2.要是可以加L（满足条件l_num<2）
转移成：f(len+1,a_num,l_num+1)
3.绝对可以加P
转移成：f(len+1,a_num,0)

结果
f(len,a_num,l_num)
中len=n; a_num<2; l_num<3
的所有的值加和

做题细节

1.定义数组之后，初始化数组的操作？
memset(f,0,sizeof f);
2.那个状态转移，不是直接=，而是要+=哦
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 len=0;len<n;len++){
            for(int a_num=0;a_num<2;a_num++){
                for(int l_num=0;l_num<3;l_num++){
                    if(a_num<1){
                        f[len+1][a_num+1][0]=(f[len][a_num][l_num]+f[len+1][a_num+1][0]) % mod;
                    }
                    if(l_num<2){
                        f[len+1][a_num][l_num+1]=(f[len][a_num][l_num]+f[len+1][a_num][l_num+1])%mod;
                    }
                    f[len+1][a_num][0]=(f[len][a_num][l_num]+f[len+1][a_num][0])%mod;
                }
            }
        }

        int res=0;
        for(int a_num=0;a_num<2;a_num++){
            for(int l_num=0;l_num<3;l_num++){
                res=(res+f[n][a_num][l_num])%mod;
            }
        }
        return res;
    }
};