LeetCode二叉树问题全解--Python版(一)

LeetCode二叉树问题全解–Python版

二叉树是由若干个节点组成，节点连接起来就成了二叉树，而节点由一个值、指向左右的两个指针组成

创建树实际上就是创建节点，然后连接节点

节点的数据类型

每个节点都有值和左右孩子，如果没有则写None。Python代码如下

class TreeNode():
    def  __init__(self, value):
        self.val = value
        self.left = None
        self.right = None

构造二叉树

这里介绍直接构造和从数组构造两种方式
以构建下图所示二叉树为例

# 例如要构造上述二叉树，则先构造节点，然后连接节点，最后返回头节点即可
def construct_tree_static():
    node = TreeNode(10)
    node2 = TreeNode(9)
    node3 = TreeNode(20)
    node4 = TreeNode(15)
    node5 = TreeNode(35)
    node.left = node2
    node.right = node3
    node3.left = node4
    node4.right = node5
    return node

# 但很多时候我们需要从数组来构造二叉树,这里特别说明数组为满二叉树显示
# 若某个节点为空，则使用None表示，例如上述二叉树的数组表述为：[10,9,20,None,None,15,35]
def construct_tree_dynamic(TreeNodeList):
    pass

树的遍历

有了树的构造，还需要树的遍历来打印每个节点的值，检查下树是否构建正确

树的遍历根据根节点输出位置的不同分为前中后序遍历三种情况

1. 先序遍历
   先访问根节点，然后访问左节点，最后访问右节点(根->左->右)
2. 中序遍历
   先访问左节点，然后访问根节点，最后访问右节点(左->根->右)
3. 后序遍历
   先访问左节点，然后访问右节点，最后访问根节点(左->右->根)

树的遍历方法可以使用递归和循环两种方式。递归的方法局限在递归层数过深时python系统压栈时会报错，所以在工业场景中一般不适用递归方式建树

树的前序遍历

# 先序遍历
# 递归方式
def pre_order_1(root):
    result = []
    if not root:#当节点为None时递归停止
        return
    result.append(root.val)
    pre_order(root.left)
    pre_order(root.right)
    print("pre_order result is %s"%",".join(result))
    return result

# 迭代方式（手动压栈），Python中的栈可以通过数组实现(这里可以自行搜索)
def pre_order_2(root):
    if not root:
        return None
    stack = [root]#根节点入栈
    

树的中序遍历

树的后序遍历