Python 数据结构与算法——二叉搜索树的实现

class Tree：本身自然需要维护根节点（root），用于指向树的第一个节点

class Tree:
root = None

class Node：每一个节点都要维护左子树、右子树

class Node:
lft = None
rgt = None

一棵二叉搜索树最为重要的两个使命是：插入和搜索，插入最终返回的是当前节点。

class Node:
lft = None
rgt = None
def __init__(self, key, val):
self.key = key
self.val = val

def insert(node, key, value):
# 返回插入后的节点
if node: return new Node(key, value)
if node.key < key:
return insert(node.rgt, key, value)
return insert(node.lft, key, value)

def search(node, key):
if node is None: raise KeyError
if node.key == key:
return node.val
if node.key < key:
return search(node.rgt, key)
return search(node.lft, key)

class Tree:
root = None
def __getitem__(self, key):
return search(self.root, key)
def __setitem__(self, key, value):
insert(self.root, key, value)
def __contains__(self, key):
try:
search(self.root, key)
return True
except KeyError:
return False