Using gen-trie
Stuff goes here
Installation
pip3 install gen-tries
Usage
Example 1 - trie of words:
from gentrie import GeneralizedTrie, TrieEntry
trie = GeneralizedTrie()
entries: list[list[str]] = [
['ape', 'green', 'apple'],
['ape', 'green'],
['ape', 'green', 'pineapple'],
]
for item in entries:
trie.add(item)
prefixes: set[TrieEntry] = trie.prefixes(['ape', 'green', 'apple'])
print(f'prefixes = {prefixes}')
suffixes: set[TrieEntry] = trie.suffixes(['ape', 'green'])
print(f'suffixes = {suffixes}')
# prefixes = {TrieEntry(ident=1, key=['ape', 'green', 'apple']),
# TrieEntry(ident=2, key=['ape', 'green'])}
# suffixes = {TrieEntry(ident=1, key=['ape', 'green', 'apple']),
# TrieEntry(ident=3, key=['ape', 'green', 'pineapple']),
# TrieEntry(ident=2, key=['ape', 'green'])}
Example 2 - trie of words used for URL search:
from gentrie import GeneralizedTrie, TrieEntry
# Create a trie to store website URLs
url_trie = GeneralizedTrie()
# Add some URLs with different components (protocol, domain, path)
url_trie.add(["https", "com", "example", "www", "/", "products", "clothing"])
url_trie.add(["http", "org", "example", "blog", "/", "2023", "10", "best-laptops"])
url_trie.add(["ftp", "net", "example", "ftp", "/", "data", "images"])
# Find all https URLs with "example.com" domain
suffixes: set[TrieEntry] = url_trie.suffixes(["https", "com", "example"])
print(suffixes)
# {TrieEntry(ident=1, key=['https', 'com', 'example', 'www', '/', 'products', 'clothing'])}
Example 3 - trie of letters from string:
from gentrie import GeneralizedTrie, TrieEntry
trie = GeneralizedTrie()
entries: list[str] = [
'abcdef',
'abc',
'abcd',
'qrf',
]
for item in entries:
trie.add(item)
suffixes: set[TrieEntry] = trie.suffixes('abcd')
print(f'suffixes = {suffixes}')
prefixes: set[TrieEntry] = trie.prefixes('abcdefg')
print(f'prefixes = {prefixes}')
# suffixes = {TrieEntry(ident=1, key='abcdef'),
# TrieEntry(ident=3, key='abcd')}
# prefixes = {TrieEntry(ident=1, key='abcdef'),
# TrieEntry(ident=3, key='abcd'),
# TrieEntry(ident=2, key='abc')}
Example 3 Output:
{TrieEntry(ident=2, key='abc'), TrieEntry(ident=1, key='abcdef')}
Example 4 - trie of numeric vectors:
from gentrie import GeneralizedTrie, TrieEntry
trie = GeneralizedTrie()
entries = [
[128, 256, 512],
[128, 256],
[512, 1024],
]
for item in entries:
trie.add(item)
suffixes: set[TrieEntry] = trie.suffixes([128])
print(f'suffixes = {suffixes}')
prefixes: set[TrieEntry] = trie.prefixes([128, 256, 512, 1024])
print(f'prefixes = {prefixes}')
# suffixes = {TrieEntry(ident=1, key=[128, 256, 512]),
# TrieEntry(ident=2, key=[128, 256])}
# prefixes = {TrieEntry(ident=1, key=[128, 256, 512]),
# TrieEntry(ident=2, key=[128, 256])}
Example 5 - trie of tuples:
from gentrie import GeneralizedTrie, TrieEntry
trie = GeneralizedTrie()
entries = [
[(1, 2), (3, 4), (5, 6)],
[(1, 2), (3, 4)],
[(5, 6), (7, 8)],
]
for item in entries:
trie.add(item)
suffixes: set[TrieEntry] = trie.suffixes([(1, 2)])
print(f'suffixes = {suffixes}')
prefixes: set[TrieEntry] = trie.prefixes([(1, 2), (3, 4), (5, 6), (7, 8)])
print(f'prefixes = {prefixes}')
# suffixes = {TrieEntry(ident=1, key=[(1, 2), (3, 4), (5, 6)]),
# TrieEntry(ident=2, key=[(1, 2), (3, 4)])}
# prefixes = {TrieEntry(ident=1, key=[(1, 2), (3, 4), (5, 6)]),
# TrieEntry(ident=2, key=[(1, 2), (3, 4)])}
Example 6 - trie used for word suggestions:
from gentrie import GeneralizedTrie, TrieEntry
trie = GeneralizedTrie()
entries: list[str] = [
'hell',
'hello',
'help',
'do',
'dog',
'doll',
'dolly',
'dolphin',
'do'
]
for item in entries:
trie.add(item)
suggestions: set[TrieEntry] = trie.suffixes('do', depth=2)
print(f'+2 letter suggestions for "do" = {suggestions}')
suggestions = trie.suffixes('do', depth=3)
print(f'+3 letter suggestions for "do" = {suggestions}')
# +2 letter suggestions for "do" = {
# TrieEntry(ident=6, key='doll'),
# TrieEntry(ident=5, key='dog'),
# TrieEntry(ident=4, key='do')}
#
# +3 letter suggestions for "do" = {
# TrieEntry(ident=6, key='doll'),
# TrieEntry(ident=5, key='dog'),
# TrieEntry(ident=4, key='do'),
# TrieEntry(ident=7, key='dolly')}
Example 7 - Checking if a key is in the trie:
from gentrie import GeneralizedTrie
trie = GeneralizedTrie()
entries: list[str] = [
'abcdef',
'abc',
'abcd',
'qrf',
]
for item in entries:
trie.add(item)
if 'abc' in trie:
print('abc is in trie')
else:
print('error: abc is not in trie')
if 'abcde' not in trie:
print('abcde is not in trie')
else:
print('error: abcde is in trie')
if 'qrf' not in trie:
print('error: qrf is not in trie')
else:
print('qrf is in trie')
if 'abcdef' not in trie:
print('error: abcdef is not in trie')
else:
print('abcdef is in trie')
# abc is in trie
# abcde is not in trie
# qrf is in trie
# abcdef is in trie