python 二叉树拷贝

说明

二叉树拷贝，就是指将一颗二叉树，复制一份

代码如下：

# 对二叉树进行拷贝
# 先对分别对左右子树进行拷贝
# 再把其作为新的树的左右子树

def copy_tree(node):
    # 如果是空树，仍然返回空树
    if node is None:
        return None
    # 分别把所有的节点递归添加到这个树中
    lt = copy_tree(node.left)
    rt = copy_tree(node.right)
    new_tree = TreeNode(node.data, lt, rt)
    return new_tree