Sentence Similarity III Python Solution

lcpy gen -n 1813   # by problem number
lcpy gen -s sentence_similarity_iii   # by problem name

Input: sentence1 = "My name is Haley", sentence2 = "My Haley"
Output: true

Input: sentence1 = "of", sentence2 = "A lot of words"
Output: false

Input: sentence1 = "Eating right now", sentence2 = "Eating"
Output: true

class Solution:
    # Time: O(n + m) where n, m are the sentence lengths
    # Space: O(n + m) for the split word lists
    def are_sentences_similar(self, sentence1: str, sentence2: str) -> bool:
        words1 = sentence1.split()
        words2 = sentence2.split()
        if len(words1) < len(words2):
            words1, words2 = words2, words1

        prefix = 0
        while prefix < len(words2) and words1[prefix] == words2[prefix]:
            prefix += 1

        suffix = 0
        while (
            suffix < len(words2) - prefix
            and words1[len(words1) - 1 - suffix] == words2[len(words2) - 1 - suffix]
        ):
            suffix += 1

        return prefix + suffix == len(words2)