add trie class

This commit is contained in:
marina 2023-08-03 16:24:41 -07:00 committed by GitHub
parent 05ffc6e0b7
commit 39262bfbe4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -18,6 +18,37 @@
<br> <br>
```python
class Trie:
def __init__(self):
self.root = {}
def insert(self, word: str) -> None:
node = self.root
for c in word:
if c not in node:
node[c] = {}
node = node[c]
node['$'] = None
def match(self, seq, prefix=False):
node = self.root
for c in seq:
if c not in node:
return False
node = node[c]
return prefix or ('$' in node)
def search(self, word: str) -> bool:
return self.match(word)
def startsWith(self, prefix: str) -> bool:
return self.match(prefix, True)
```
<br>
---- ----
### insertion ### insertion