继承方法的正确方法有1个更正

Question

我在做数据结构的家庭作业时用python做了一点面向对象的实验，我在理解如何用修正继承一些方法时遇到了一些困难

所以，我有：

class BinarySearchTree(object):
    # snipped
    def _insert(self, key, val):
        prevNode = None
        currentNode = self.root
        while currentNode:
            prevNode = currentNode
            if key < currentNode.key:
                currentNode = currentNode.leftChild
            else:
                currentNode = currentNode.rightChild

        if key < prevNode.key:                      
            prevNode.leftChild = self.Node(key, val, parent=prevNode)
            self.updateNodeProperties(prevNode.leftChild)            
        else:                                       
            prevNode.rightChild = self.Node(key, val, parent=prevNode)
            self.updateNodeProperties(prevNode.rightChild)
    # snipped                

以及：

class RBTree(BinarySearchTree): 
    # snipped
    def _insert(self, key, val):
        prevNode = None
        currentNode = self.root
        while currentNode:
            prevNode = currentNode  
            prevNode.size += 1 # The only difference is in this line
            if key < currentNode.key:
                currentNode = currentNode.leftChild
            else:
                currentNode = currentNode.rightChild

        if key < prevNode.key:                      
            prevNode.leftChild = self.Node(key, val, parent=prevNode)
            self.updateNodeProperties(prevNode.leftChild)            
        else:                                       
            prevNode.rightChild = self.Node(key, val, parent=prevNode)
            self.updateNodeProperties(prevNode.rightChild)
    # snipped

我的问题是：有没有一种聪明的方法来继承这个方法并实现这个差异（prevNode.size += 1），而不复制整个继承的函数代码

p.S.：对不起，我的英语不好

UPD：现在我无法在斯科特和卡布的解决方案之间做出选择

Answer 1

这里有一个使用访问者设计模式的建议

在这种模式中，您正在编写一个visitor方法，而不知道访问的每个节点需要做什么

我们将更改基类以添加访问者功能

class BinarySearchTree(object):
# snipped
def _insert(self, key, val, visitor=None):   # Change here
    prevNode = None
    currentNode = self.root
    while currentNode:
        prevNode = currentNode
        if visitor: visitor(prevNode)       # Change here
        if key < currentNode.key:
            currentNode = currentNode.leftChild
        else:
            currentNode = currentNode.rightChild

    if key < prevNode.key:                      
        prevNode.leftChild = self.Node(key, val, parent=prevNode)
        self.updateNodeProperties(prevNode.leftChild)            
    else:                                       
        prevNode.rightChild = self.Node(key, val, parent=prevNode)
        self.updateNodeProperties(prevNode.rightChild)
# snipped   

现在是你的第二节课

class RBTree(BinarySearchTree): 
# snipped
def _insert(self, key, val):
    visitor = lambda node: node.size += 1
    super(RBTree, self)._insert(self, key, val, visitor)
# snipped