python - 向二叉搜索树添加一个元素

Question

您好，我的问题是如何修复我的二叉搜索树程序的添加函数的代码。

class BTNode2:
    def __init__(self,d,l,r):
        self.data = d
        self.left = l
        self.right = r
        self.mult = 1

这是添加方法

def add(self, d):   
    if (self < d.data):
        if (d.left != None):
            add(self, d.left)
        else:
            d.left = BTNode2(self)
    else:
        if (d.right != None):
            add(self, d.right)
        else:
            d.right = BTNode2(self)
    return 

这是我尝试运行添加方法时遇到的错误:

AttributeError: 'str' 对象没有属性 'data'

Answer 1

本质上，您交换了 add 函数的参数，因为 self 是这里的树，而 d 是要添加的数据元素.此外，为了构造只有一个参数的 BTNode2，您应该为 l 和 r 添加默认值。最后，根据 mult 的作用，您可能想在 add 算法中更改它，但它代表什么并不是很清楚。

所以我们可以解决这个问题:

class BTNode2:
    def __init__(self, d, l<b>=None</b>, r<b>=None</b>):
        self.data = d
        self.left = l
        self.right = r
        self.mult = 1

    def add(self, d):   
        if d < self.data:
            if self.left is not None:
                <b>self.left.add(d)</b>
            else:
                self.left = BTNode2(d)
        else:
            if self.right is not None:
                <b>self.right.add(d)</b>
            else:
                self.right = BTNode2(d)