【LeetCode】552.学生出勤记录

最新推荐文章于 2024-09-28 10:34:40 发布

陈言之、

最新推荐文章于 2024-09-28 10:34:40 发布

阅读量671

点赞数 16

分类专栏： LeetCode 文章标签： leetcode 算法

本文链接：https://blog.csdn.net/qq_45427698/article/details/141335461

版权

LeetCode 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目：

可以用字符串表示一个学生的出勤记录，其中的每个字符用来标记当天的出勤情况（缺勤、迟到、到场）。记录中只含下面三种字符：
‘A’：Absent，缺勤
‘L’：Late，迟到
‘P’：Present，到场
如果学生能够同时满足下面两个条件，则可以获得出勤奖励：
按总出勤计，学生缺勤（‘A’）严格少于两天。
学生不会存在连续 3 天或连续 3 天以上的迟到（‘L’）记录。
给你一个整数 n ，表示出勤记录的长度（次数）。请你返回记录长度为 n 时，可能获得出勤奖励的记录情况数量。答案可能很大，所以返回对 $10^9 + 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 <= $10^5$

思考

首先看到这个题目我们用最笨的方法，我直接递归走。
即我们以n为结束点，m来记录我们连续的迟到数目，用a来表示我们缺勤的总数。
通过题目我们知道，我们缺勤的数目最多为1，迟到数目连续的走最多为2，那么它就是我们的限制。
所以代码如下：

class Solution {
    // 其中n表示的是出勤的长度，m表示的是连续迟到的次数，a表示的是缺勤的次数
    private int res = 0;
    private void cycle(int n,int m,int a){
        if(n<=0){
            res = (res + 1)%(1000000007);
            return;
        }
            
        // 即该条件下可以为迟到,即最多迟到两次
        if(m<2){
            m += 1;
            cycle(n-1,m,a);
        }
        m = 0;
        if(a<1){
            a += 1;
            cycle(n-1,m,a);
            a -= 1;
        }
        cycle(n-1,m,a);
        return;
    }
    public int checkRecord(int n) {
        // 当n=1时，无论其为缺勤、迟到、到场都可以拿到奖励
        // 当n=2时，因为缺勤必须少于2天，则A只能为1，迟到少于连续的3天，则可以全为迟到，
        // 当n=3时，可以根据前者L的记录来计算该位置是否可以为迟到L
        // ...
        // 那么我们如果用遍历走的话
        cycle(n,0,0);
        return res;
    }
}

很显然,超时。
那么我们怎么去优化呢？用空间去节省时间，我们动态规划走。
dp[i][j][k]来表示我们的状态，其中i表示前i天，j表示的是我们前i天的缺勤数目（0，1），k表示我们前i天，即结尾连续迟到的数目，即0，1，2。
而我们前i天的可奖励的出勤记录的数量是根据前i-1来确定的.

如果我们第i天是P，那么我们第i-1天可以是dp[i-1][0][0],dp[i-1][0][1],dp[i-1][0][2],dp[i-1][1][0],dp[i-1][1][1],dp[i-1][1][2].
如果我们第i天是A，那么我们第i-1天可以是dp[i-1][0][0],dp[i-1][0][1],dp[i-1][0][2]
如果我们第i天是L，那么我们第i-1天可以是dp[i-1][0][0],dp[i-1][0][1],dp[i-1][1][0],dp[i-1][1][1]
综上所述，我们的代码如下：

class Solution {
    
    public int checkRecord(int n) {
        // 当n=1时，无论其为缺勤、迟到、到场都可以拿到奖励
        // 当n=2时，因为缺勤必须少于2天，则A只能为1，迟到少于连续的3天，则可以全为迟到，
        // 当n=3时，可以根据前者L的记录来计算该位置是否可以为迟到L
        // ...
        // 找规律，如果我们最差的情况，即LLA走，那么有2种排序。LLA,ALL
        final int MOD = 1000000007;
        int[][][] dp = new int[n+1][2][3];
        dp[0][0][0] = 1;
        for(int i = 1;i<=n;i++){
            // 以P结尾的数量
            for(int j = 0;j<=1;j++){
                for(int k=0;k<=2;k++){
                    dp[i][j][0] = (dp[i][j][0] + dp[i-1][j][k]) % MOD;
                }
            }
            // 以A结尾
            for(int k = 0;k<=2;k++){
                dp[i][1][0] = (dp[i][1][0] + dp[i-1][0][k]) % MOD;
            }
            // 以L结尾
            for(int j = 0;j<=1;j++){
                for(int k = 1;k<=2;k++){
                    dp[i][j][k] = (dp[i][j][k]+dp[i-1][j][k-1])%MOD;
                }
            }
        }
        int sum = 0;
        for(int j = 0;j<=1;j++){
            for(int k = 0;k<=2;k++){
                sum = (sum + dp[n][j][k])%MOD; 
            }
        }
        return sum;
    }
}

当然再此基础上我们还可以简化空间复杂度，因为我们的dp[i][j][k]是由前一天所决定，而我们在计算前i天时，我们的前i天还没计算出来，因而我们的dp[j][k]在未更新前是表示前i-1天的情况的。此代码忽略，背包哪里也有讲。
最后我看人家是用矩阵快速幂走的，忘记具体咋用了，等下次复习到了再说吧。