具有键和值的 Python 二叉搜索树

Question

我需要实现一个二叉搜索树类作为作业，但我很难实现插入功能。我已经通过谷歌看了很多，以找到一些解决方案或可能性，但他们都没有使用键和值（主要是值），或者如果他们也使用了键，他们有很多单独的功能，我是我认为不允许这样做。

所以预建就是：

class Node:
    def __init__(self, key, value):
        self.key = key
        self.value = value
        self.left = self.right = None

class BinarySearchTree:
    def __init__(self):
        self.root = None
        self.size = 0

    def __len__(self):
        return self.size

    def insert(self, key, value):
        pass

    def remove(self, key):
        pass

    def find(self, key):
        pass

现在的问题是，如果我想检查例如该值是小于还是大于当前节点以将其放在右侧或左侧，我会收到诸如“root is not defined”或“root.right”之类的错误" 没有这样的属性等... 我想这是有道理的，因为 self.root 被声明为 None。

但是我现在如何真正修复它以使插入功能正常工作？

我对这个任务有点困惑，因为它使用键+值，所以我需要插入绑定到特定键的值，如果键已经存在，覆盖它的值。

Answer 1

它是早上 5 点，所以这可能是错误的，但这里是：
key 是我们排序的依据，值不感兴趣
你的插入函数应该看起来像这样：

def insert(self, key, value):
        if self.root = None:
            self.root = Node(key,value)
            return
        #regular binary tree traversal (comparing the key) to find where to insert, lets assume we need to insert on the left
        parent.left = Node(key,value)

你能从这里弄清楚还是你想要更多的方向

Answer 2

您没有指定，但我猜测键的目的是确定特定键是否已经在树中，如果是，则替换 O(1) 运行时复杂度中的相关节点的值。

因此，当您插入节点时，您将首先检查字典中的键（您将在__init__ 中自己初始化一个空字典）。如果它已经存在，那么您只需要替换该特定键的节点值即可。否则，您添加新节点的方式与在任何 BST 中相同，并且还记得更新您的字典以将键映射到它的节点。