Leetcode: Student Attendance Record I

You are given a string representing an attendance record for a student. The record only contains the following three characters:
'A' : Absent.
'L' : Late.
'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

 

1-liner

s.contains("") normally is O(nm), but can be optimized to be O(n)

public class Solution {
    public boolean checkRecord(String s) {
        if(s.indexOf("A") != s.lastIndexOf("A") || s.contains("LLL"))
            return false;
        return true;
    }
}

 

O(n) scan

class Solution {
    public boolean checkRecord(String s) {
        int countA = 0, countB = 0;
        for (char c : s.toCharArray()) {
            switch (c) {
                case 'A': 
                    if (countA == 1) return false;
                    countA ++;
                    countB = 0;
                    break;
                case 'L':
                    if (countB == 2) return false;
                    countB ++;
                    break;
                default:
                    countB = 0;
            }
        }
        return true;
    }
}