leetcode

Leetcode Solutions (0x6A73)
git clone https://git.pngu.org/leetcode
Log | Files | Refs | Submodules | README

check_word_occur_prefix_sentence.py (1171B)


# 1455. Check If a Word Occurs As a Prefix of Any Word in a Sentence

"""
given a 'sentence' that consists of some words separated by a single space,
and a 'searchWord', check if 'searchWord' is a prefix of any word in
'sentence'. return the index of the word in 'sentence' (1-indexed) where
'searchWord' is a prefix of the word. if 'searchWord' is a prefix of more
than one word, return the index of the first word (minimum index). if there
is no such word, return -1. a prefix of a string 's' is any leading
contiguous substring of 's'.
"""


class Solution(object):
    def isPrefixOfWord(self, sentence, searchWord):
        """
        :type sentence: str
        :type searchWord: str
        :rtype: int
        """
        for i, w in enumerate(sentence.split(" "), 1):
            if w.startswith(searchWord):
                return i
        return -1


if __name__ == "__main__":
    obj = Solution()
    print(obj.isPrefixOfWord(sentence="i love eating burger", searchWord="burg"))
    print(
        obj.isPrefixOfWord(sentence="this problem is an easy problem", searchWord="pro")
    )
    print(obj.isPrefixOfWord(sentence="i am tired", searchWord="you"))