[数据结构][Python][经典题目]二叉搜索树

最新推荐文章于 2020-11-23 23:39:32 发布

Twish

最新推荐文章于 2020-11-23 23:39:32 发布

阅读量125

点赞数

分类专栏： Python Algorithm python数据结构

本文链接：https://blog.csdn.net/qq_27695659/article/details/90730295

版权

Python 同时被 3 个专栏收录

283 篇文章 1 订阅

订阅专栏

python数据结构

31 篇文章 1 订阅

订阅专栏

Algorithm

27 篇文章 0 订阅

订阅专栏

class Node:
    lft = None
    rgt = None
    def __init__(self,key,val):
        self.key = key
        self.val = val
def insert(node,key,val):
    if node is None: return Node(key,val)
    if node.key == key: node.val = val
    elif key < node.key:
        node.lft = insert(node.lft, key, val)
    else:
        node.rgt= insert(node.rgt,key,val)
    return node
def search(node,key):
    if node is Node: raise KeyError
    if node.key == key: return node.val
    elif key < node.key:
        return search(node.lft,key)
    else:
        return search(node.rgt,key)

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