1065. Index Pairs of a String
Содержание
class Trie:
def __init__(self):
self.children = [None] * 26
self.is_end = False
def insert(self, word):
node = self
for c in word:
idx = ord(c) - ord('a')
if node.children[idx] is None:
node.children[idx] = Trie()
node = node.children[idx]
node.is_end = True
class Solution:
def indexPairs(self, text: str, words: List[str]) -> List[List[int]]:
trie = Trie()
for w in words:
trie.insert(w)
n = len(text)
res = []
for i in range(n):
node = trie
for j in range(i, n):
idx = ord(text[j]) - ord('a')
if node.children[idx] is None:
break
node = node.children[idx]
if node.is_end:
res.append([i, j])
return res