LeetCode: 551. 学生出勤记录 I

551. 学生出勤记录 I

原题

给你一个字符串 s 表示一个学生的出勤记录，其中的每个字符用来标记当天的出勤情况（缺勤、迟到、到场）。记录中只含下面三种字符：

• 'A'：Absent，缺勤
• 'L'：Late，迟到
• 'P'：Present，到场

如果学生能够 同时 满足下面两个条件，则可以获得出勤奖励：

• 按 总出勤 计，学生缺勤（'A'）严格 少于两天。
• 学生 不会 存在 连续 3 天或 连续 3 天以上的迟到（'L'）记录。

如果学生可以获得出勤奖励，返回 true ；否则，返回 false 。

示例 1：

输入：s = "PPALLP"
输出：true
解释：学生缺勤次数少于 2 次，且不存在 3 天或以上的连续迟到记录。

示例 2：

输入：s = "PPALLL"
输出：false
解释：学生最后三天连续迟到，所以不满足出勤奖励的条件。

提示：

• 1 <= s.length <= 1000
• s[i] 为 'A'、'L' 或 'P'

class Solution {
    public boolean checkRecord(String s) {
				
    }
}

解题思路

1. 创建两个计数器：缺勤天数和连续迟到天数。
2. 遍历字符数组：
   • 如果缺勤天数少于 2 且连续迟到天数少于 3，则继续执行。
   • 如果是出勤（‘P’），重置连续迟到天数为 0。
   • 如果是迟到（‘L’），连续迟到天数加 1。
   • 如果是缺勤（‘A’），重置连续迟到天数为 0，增加缺勤天数。
   • 如果缺勤天数大于等于 2 或连续迟到天数大于等于 3，则返回 false。

代码示例

class Solution {
    public boolean checkRecord(String s) {
        // 将字符串转换成字符数组
        char arr[] = s.toCharArray();

        // 记录缺勤的天数
        int absentCount = 0;

        // 记录连续迟到的天数
        int continuousLateCount = 0;

        // 一次遍历整个字符数组
        for (char ch : arr) {
          	// 缺勤次数少于 2 且连续迟到次数少于 3 才执行
            if (absentCount < 2 && continuousLateCount < 3) {
                if (ch == 'P') {
                  	// 遇到非迟到的出勤记录清空「连续迟到」的计数
                    continuousLateCount = 0;
                } else if (ch == 'L') {
                  	// 连续迟到天数的计数加 1
                    continuousLateCount += 1;
                } else {
                  	// 遇到非迟到的出勤记录清空「连续迟到」的计数
                    continuousLateCount = 0;
                  	// 缺勤天数加 1
                    absentCount += 1;
                }
            } else {
                return false;
            }
        }

        return absentCount < 2 && continuousLateCount < 3;
    }
}

注意

整个字符数组遍历完成之后不能直接 return true，考虑诸如 s = "PPALLL"、s = "LLLAA"的情况，最后一个字符为 L 或 A，刚好达到连续三天迟到或者两天缺勤，此时 if 判断 if (absentCount < 2 && continuousLateCount < 3) 不会再被执行，因此要再一次进行判断后返回。

优化

使用字符索引代替字符数组

从空间复杂度的角度来考虑，char arr[] = s.toCharArray() 将字符串转换成字符数组需要额外的内存开销。可以修改成如下代码，直接根据字符的索引来迭代：

class Solution {
    public boolean checkRecord(String s) {
        // ...

        // 根据字符的索引迭代
        for (int i = 0; i < s.length(); i++) {
          	// 获取字符
          	char ch = s.charAt(i);
						
            if (absentCount < 2 && continuousLateCount < 3) {
                if (ch == 'P') {
          				// ...
                }
            }
        }
      	
      	return absentCount < 2 && continuousLateCount < 3;
    }
}

使用字符串方法 contains() 减少手动计数的开销

直接对字符串调用 contains() 方法来判断是否存在连续三天的迟到情况，即 s.contains("LLL")，删去多余手动记录连续迟到天数的内存开销。

class Solution {
    public boolean checkRecord(String s) {
        // 记录缺勤的天数
        int absentCount = 0;

      	// 判断缺勤天数是否大于 2
        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i) == 'A' && ++absentCount > 1) {
                return false;
            }
        }

        // 判断是否出现连续三天迟到的情况
        return !s.contains("LLL");
    }
}

说明

if (s.charAt(i) == 'A' && ++absentCount > 1)：

• 遇到字符 A 时 s.charAt(i) == 'A' 条件成立，继续执行后面的判断。++absentCount 会先执行递增操作，然后与 1 进行比较，简化了代码。

• 遇到非字符 A 时 s.charAt(i) == 'A' 条件不成立，&& 逻辑与运算符直接短路，将不会执行第二个操作数。