aboutsummaryrefslogtreecommitdiff
path: root/check_word_occur_prefix_sentence.py
blob: aa16f70ea602ab0e6b5a8b6673913d34395e89aa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# 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"))