occurrences-after-bigram.py

class Solution:
    def findOcurrences(self, text: str, first: str, second: str) -> List[str]:
        res = []
        text = text.split()
        for i in range(2, len(text)):
            if text[i - 2] == first and text[i - 1] == second:
                res.append(text[i])
        return res