python算法学习——二叉树

二叉树的结构

二叉树特点是每个结点最多只能有两棵子树，且有左右之分。

二叉树的特殊类型：
满二叉树：
如果一棵二叉树只有度为0的结点和度为2的结点，并且度为0的结点在同一层上，则这棵二叉树为满二叉树；
完全二叉树：
深度为k，有n个结点的二叉树当且仅当其每一个结点都与深度为k的满二叉树中编号从1到n的结点一一对应时，称为完全二叉树。
二叉搜索树：
若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值； 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值； 它的左、右子树也分别为二叉排序树。二叉搜索树作为一种经典的数据结构，它既有链表的快速插入与删除操作的特点，又有数组快速查找的优势；所以应用十分广泛。
python中二叉树节点定义如下，含有值val，左节点left，右节点right：

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

LeetCode上的二叉树表题目分享

二叉树主要是通过遍历实现的，一般需要用到递归或者迭代，对于递归，需要注意其需要递归之处及终止条件。

1. 剑指offer 25 二叉树的深度

重点在于递归原理是，主树的深度为其左右子树中最大值加一，因此根据此原理构建处maxDepth函数。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        if not root: return 0 
        else: return max(self.maxDepth(root.right),self.maxDepth(root.left))+1

2. 剑指offer 55 平衡二叉树

解法一：常规方法，先计算出左右子树的深度再进行比较。
不过这样会导致很多节点的重复计算

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        def depth(root:TreeNode):	# 计算深度
            if not root: return 0
            return max(depth(root.right),depth(root.left))+1
        # 判断，因为要判断左右子树本身是否为平衡树
        if not root: return True
        if abs(depth(root.left)-depth(root.right)) <=1 and self.isBalanced(root.left) and self.isBalanced(root.right): return True
        return False

解法二：
参考解法
优化的地方在于，事先判断左子树和右子树是否为平衡树，如果不是则直接退出（返回-1），达到了剪枝的目的。

        # 方法二
        def find(root:TreeNode):
            if not root: return 0
            left = find(root.left)  # -1
            if left == -1:return -1
            right = find(root.right)    # -1
            if right == -1:return -1
            return max(left,right)+1 if abs(left-right)<=1 else -1
        return find(root) != -1

3. 剑指offer 28 对称二叉树

重点在于迭代的停止条件，
若两边节点为空，则返回True，若有一个为空，或者均不为空，但值不同，则返回False；
迭代部分是对左子树的左节点和右子树的右节点，以及左子树的右节点和右子树的左节点进行比较。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isSymmetric(self, root: TreeNode) -> bool:
        def duicheng(R,L):
            # 两边同时为空
            if not L and not R: return True
            # 若有一个为空，或者都不为空，但值不同
            if not L or not R or (L.val != R.val): return False
            return duicheng(R.left,L.right) and duicheng(L.left,R.right)
        return duicheng(root.right,root.left) if root else True

4. 剑指offer 27 二叉树的镜像

重点在于交换一个节点的左右子节点。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def mirrorTree(self, root: TreeNode) -> TreeNode:
        if not root: <