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"))
|