python 前缀树实现

class Trie:
    def __init__(self):
        self.root = dict()
        self.end = "$"

    def insert(self, item: str):
        node = self.root
        for char in item:
            node = node.setdefault(char, {})  # 有值 return值; 无值设值并 return值
        node[self.end] = self.end

    def __collection(self, node):
        if node == self.end:
            return [""]
        res = []

        for i in node:
            the = node[i]
            remain = self.__collection(the)
            for j in remain:
                res.append(i + j)

        return res

    def __getitem__(self, item: str):
        node = self.root
        for char in item:
            if char not in node:
                return []
            node = node[char]
        output = self.__collection(node)
        for i in range(len(output)):
            output[i] = (item + output[i])[:-1]
        return output

    def reset(self):
        self.root.clear()