【LeetCode】二叉树的最近公共祖先和树的子结构

题目描述

1. （二叉树的最近公共祖先）给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
   百度百科中最近公共祖先的定义为：“对于有根树 T 的两个节点 p、q，最近公共祖先表示为一个节点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”
2. （树的子结构）输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
   B是A的子结构， 即 A中有出现和B相同的结构和节点值。

方法（参考）

前序遍历+递归

代码一（236.二叉树的最近公共祖先）

二叉树的最近公共祖先中，两个指定节点的最近公共祖先一定满足下面三个条件之一：
1）两个指定节点分布在当前节点的左子树和右子树中，则当前节点即为最近公共祖先；
2）最近公共祖先是两个指定节点之一，则对应另外一个指定节点一定在其左子树或右子树中。
对二叉树进行先序遍历，那么返回的最近公共祖先的判定条件：
1）当当前节点为空或等于两个指定节点之一时，返回当前节点：

        if not root or root == p or root == q:
            return root

2）若左节点为空，右节点不为空，则右节点为公共祖先：

        if not left:
            return right

3）若右节点为空，左节点不为空，则左节点为公共祖先：

        if not right:
            return left

4）若左右节点对应指定节点，则当前节点即为公共祖先：

		return root

总代码

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

class Solution(object):
    def lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        if not root or root == p or root == q:
            return root
        left = self.lowestCommonAncestor(root.left, p, q)
        right = self.lowestCommonAncestor(root.right, p, q)
        if not left:
            return right
        if not right:
            return left
        return root

代码二（剑指 Offer 26. 树的子结构）

树的子结构中，采用先序遍历查找子结构是否存在。

1. 当 A和B树有一个为null时返回False;
2. 然后依次在当前节点，以及当前节点的左子树和右子树中查找子结构。若存在则直接返回True.

对于查找函数，通样采用先序遍历

1） b为空，说明b的结构以及查找完,是a的子结构，则返回True.
2） a为空或者a节点的值不等于b节点的值，则说明节点不匹配则返回False
3） 如果a的左子树和b的左子树匹配完并且a的右子树和b的右子树匹配完，则返回True.

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

class Solution(object):
    def isSubStructure(self, A, B):
        """
        :type A: TreeNode
        :type B: TreeNode
        :rtype: bool
        """
        def recur(a, b):
            if not b:
                return True
            if not a or a.val != b.val:
                return False
            return recur(a.left, b.left) and recur(a.right, b.right)
        return bool(A and B) and (recur(A, B) or self.isSubStructure(A.left, B) or self.isSubStructure(A.right, B))