Implementation of trie

trie.py

from collections import defaultdict

class TrieNode:
    def __init__(self):
        self.isWord = False
        self.children = defaultdict(TrieNode)


class Trie:
    def __init__(self):
        self.root = TrieNode()

    def insert(self, word):
        node = self.root
        for c in word:
            node = node.children[c]
        node.isWord = True