Tricks（二十三）—— Python 实现树这种数据结构

普通二叉树

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

当然也在构造时，传递进来左子树/右子树，如下：

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

可回溯的二叉树

通过维护一个指向父节点的指针实现回溯：

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

此时若想维护，父节点与子节点的互关系（父到子，子到父），需定义如下的成员函数：

def set_left(self, left):
    self.left = left
    left.parent = self
def set_right(self, right):
    self.right = right
    right.parent = self

• （1）根节点（root）的父节点为空；
• （2）叶节点（leaf）的左子树节点（left）右子树节点，均为空；
• （3）内部节点（internal）的左右子树节点不全为空，父节点不为空；

树功能的拓展

对树功能的拓展，即是引入新的成员变量，以及由此展开的成员函数，例如想在树的迭代构建的过成中，当前节点的深度（depth）：

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

depth 参数也是在迭代的过程中进行的指定。