python中树的运用样例

目录

一、文件系统样例

二、Trie树

一、文件系统样例

class FileNode:def __init__(self, name, is_file=False):self.name = nameself.is_file = is_fileself.children = []def add_child(self, child):self.children.append(child)# 创建文件系统结构
root = FileNode("root")
home = FileNode("home")
user = FileNode("user")
file1 = FileNode("file1.txt", True)
file2 = FileNode("file2.txt", True)root.add_child(home)
home.add_child(user)
user.add_child(file1)
user.add_child(file2)def print_tree(node, level=0):print(" " * level + node.name)for child in node.children:print_tree(child, level + 2)print_tree(root)

二、Trie树

class TrieNode:def __init__(self):self.children = {}self.is_end_of_word = Falseclass Trie:def __init__(self):self.root = TrieNode()def insert(self, word):node = self.rootfor char in word:if char not in node.children:node.children[char] = TrieNode()node = node.children[char]node.is_end_of_word = Truedef search(self, word):node = self.rootfor char in word:if char not in node.children:return Falsenode = node.children[char]return node.is_end_of_word# 使用Trie树
trie = Trie()
trie.insert("hello")
trie.insert("world")print(trie.search("hello"))  # 输出 True
print(trie.search("world"))  # 输出 True
print(trie.search("helloo"))  # 输出 False