trie.py

from typing import Any, Dict

class Trie:
    def __init__(self) -> None:
        self.children : Dict[str, Trie] = {}
        self.value : Any = None
    
    def find(self, key: str) -> Any:
        node = self
        for char in key:
            if char in node.children:
                node = node.children[char]
            else:
                return None
        return node.value
    
    def insert(self, key: str, value: Any) -> None:
        node = self
        for char in key:
            if char not in node.children:
                node.children[char] = Trie()
            node = node.children[char]
        node.value = value