Question

我是Python和数据结构的新手。在学习二叉树时，我发现所有可用的实现都在节点类中包含了树的方法。

但是我想以不同的方式实现它，与Node类一起，将有一个Binary树类，其中将包含Binary树的方法。

这是我想出的代码，但是它给我传递给函数insertleft()和insertright()的参数带来了错误。

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

class Bt():
    def __init__(self, root):
        self.root = root

    def insertleft(temp.left, data):    
        if (temp.left == None):
            temp.left.data = data
        elif(data<temp.left.data):
            t1 = temp.left
            insertleft(t1.left, data)
        elif(data>temp.left.data):
            t1 = temp.left
            insertright(t1.left, data)

    def insertright(temp.right, data):  
        if (temp.right == None):
            temp.right.data = data
        elif(data<temp.right.data):
            t1 = temp.right
            insertleft(t1.right, data)
        elif(data>temp.right.data):
            t1 = temp.right
            insertright(t1.right, data)


    def insert(self, data):
        temp = self.root
        if(temp.data = None):
            temp.data = data
        elif(data<temp.data):
            insertleft(temp.left, data)
        elif(data>temp.data):
            insertright(temp.right, data)


r = Bt()
r.root = Node(5)
r.insert(4)
r.insert(6)

这些是我收到的错误：

def insertleft(temp.left, data):  
                      ^
SyntaxError: invalid syntax

我不确定正确的语法应该是什么。预先感谢您的帮助

Answer 1

定义一个函数时，您还定义要接受的参数。但是，在类内部，只要方法不是静态的或类方法，则第一个参数应为self或代表实例对象本身的变量。

对于您来说，在调用函数时已经传递了temp.left和temp.right的值。将函数定义更改为仅使用temp应该可以正常工作。甚至更易读，node，因为这实际上是您所指的。

您的insertleft和insertright函数的作用与insert相同，并且不需要包含它们。只需修改insert函数即可完成所有操作。

这是最终结果应该类似于：

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

class Bt():
    def __init__(self, root=None):  #Added default value
        self.root = Node(root)

    def insert(self, data):
        if self.root.data is None:
            self.root.data = data   #Set the root data if it doesn't exist.
        else:
            self._insert(self.root, data)   #Prime the helper function with root node

    def _insert(self, node, data):    #Insert helper function to do the insertion
        if data < node.data:    #Check if data needs to be inserted on the left
            if node.left is None:   #Set left node if it doesn't exist
                node.left = Node(data)
            else:                   #Else let the left node decide where it goes as a child
                self._insert(node.left, data)
        else:                   #Else data needs to be inserted on the right
            if node.right is None:  #Set right node if it doesn't exist
                node.right = Node(data)
            else:                   #Else let the right node decide where it goes as a child
                self._insert(node.right, data)


r = Bt(5)
r.insert(4)
r.insert(6)

现在您只是缺少打印树和数据...以及访问数据的功能。

二叉树实现，节点和树具有单独的类

1 个答案: