1032_streamChecker.py

# 1032. Stream of Characters

# Anish Malla's Solution: https://www.youtube.com/watch?v=Z9J2SEORBug
# https://leetcode.com/problems/stream-of-characters/discuss/807900/Python-Simple-Trie-Solution-Explained-(code-%2B-video

class StreamChecker:

    def __init__(self, words: List[str]):
        self.trie = {}
        self.nodes = []
        for word in words:
            node = self.trie
            for char in word:
                if char not in node:
                    node[char] = {}
                node = node[char]
            node['end'] = True
        

    def query(self, letter: str) -> bool:
        
        self.nodes.append(self.trie)
        temp = False
        new_nodes = []
        
        for node in self.nodes:
            if letter in node:
                node = node[letter]
                if 'end' in node:
                    temp = True
                new_nodes.append(node)
        self.nodes = new_nodes
        return temp

# Your StreamChecker object will be instantiated and called as such:
# obj = StreamChecker(words)
# param_1 = obj.query(letter)