551. Student Attendance Record I -- Python

551. Student Attendance Record I

You are given a string representing an attendance record for a student. The record only contains the following three characters:
1. ‘A’ : Absent.
2. ‘L’ : Late.
3. ‘P’ : Present.
A student could be rewarded if his attendance record doesn’t contain more than one ‘A’ (absent) or more than two continuous ‘L’ (late).

You need to return whether the student could be rewarded according to his attendance record.

Example 1:

Input: "PPALLP"
Output: True

Example 2:

Input: "PPALLL"
Output: False

思路：

先判断A的个数就可以了，然后就是看有没有“L”的连续超过2个的。（”LLALL“ 返回是True,”LLALLL”才返回 False）

代码：

class Solution:
    def checkRecord(self, s):
        """
        :type s: str
        :rtype: bool
        """
        if s.count("A")>1:
            return False
        if s.count("L")>2:
            for i in range(2,s.count("L")):
                if s.count("L"* (i+1)) != 0:
                    return False
            return True
        else:
            return True