python 在二叉樹中搜索節點

Question

我在二叉樹中搜索時遇到問題，我想檢查根是否存在但遇到錯誤。 我究竟做錯了什么？

這是我的完整代碼：

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

    def printTree(self):
        if self.left:
            self.left.PrintTree()
        print(self.data),
        if self.right:
            self.right.PrintTree()

    def insert(self, data):
        """ Compare the new value with the parent node """
        if self.data:
            if data < self.data:
                if self.left is None:
                    self.left = Node(data)
                else:
                    self.left.insert(data)
            elif data > self.data:
                if self.right is None:
                    self.right = Node(data)
                else:
                    self.right.insert(data)
        else:
            self.data = data

    def is_exist(self, val):
        if val < self.data:
            if self.left is None:
                return None, None
            return self.left.exists(val, self)
        elif val > self.data:
            if self.right is None:
                return None, None
            return self.right.exists(val, self)
        else:
            return self.data

這是我的搜索 function 以檢查是否存在 root

def is_exist(self, val): if val < self.data: if self.left is None: return None, None return self.left.exists(val, self) elif val > self.data: if self.right is None : return None, None return self.right.exists(val, self) else: return self.data

這是測試：

def test_binary_tree():
    root = Node(10)
    assert root.is_exist(10)
    root.insert(4)
    assert root.is_exist(4)
    root.insert(11)
    assert root.is_exist(11)
    root.insert(3)
    assert root.is_exist(3)
    root.insert(770)
    assert root.is_exist(770)
    root.insert(523)
    assert root.is_exist(523)
    root.insert(43)

print(test_binary_tree())

我得到的錯誤：

    return self.left.exists(val, self)
AttributeError: 'Node' object has no attribute 'exists'

Answer 1

您用錯誤的名稱稱呼您的 function。 此外，如果您調用對象的方法，則不會指定self 。 見這里。 你寫了return self.right.exists(val, self)但你應該寫了return self.right.is_exist(val)