剑指offer的Python实现之二叉树(2)：相同的树

相同的树及其相关题目

目录

相同的树及其相关题目

1. 相同的树

2. 对称的二叉树

3.  树的子结构

---

1. 相同的树

题目描述：

        给定两个二叉树，编写一个函数来检验它们是否相同。如果两个树在结构上相同，并且节点具有相同的值，则认为它们是相同的。

class Solution(object):
    def isSameTree(self, p, q):
        """
        :type p: TreeNode
        :type q: TreeNode
        :rtype: bool
        """
        if p==None and q==None:
            return True
        if p and q and p.val==q.val:
            return self.isSameTree(p.left,q.left) and self.isSameTree(p.right, q.right)
        return False

2. 对称的二叉树

题目描述：

        请实现一个函数，用来判断一颗二叉树是不是对称的。注意，如果一个二叉树同此二叉树的镜像是同样的，定义其为对称的。

解题思路：

       

        由上图可知，一颗对称的二叉树，其左树与右树互为镜像（二叉树的镜像详见上一篇文章），即左树的左子树等于右树的右子树。因此，可以将树的左树和右数看作是两棵树，解题方法与上面一题“相同的树”相同，只是将判断两棵树是否相同换成一棵树的左子树是否与另一棵树的右子树相同。

class Solution(object):
    def isSymmetric(self, root):
        """
        :type root: TreeNode
        :rtype: bool
        """
        if not root:
            return True
        return self.isSame(root.left,root.right)
    
    def isSame(self, p, q):
        if not p and not q:
            return True
        if p and q and p.val==q.val:
            return self.isSame(p.left,q.right) and self.isSame(p.right, q.left)
        return False

3.  树的子结构

题目描述：

        输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）。

解题思路：

                             

                       两颗二叉树A和B，右边的树B是左边的树A的子结构

        这道题其实就是“二叉树的遍历”与“相同的树两道题的结合”，首先遍历二叉树A，在A的每个节点判断该节点下是否与B结构相同的树。

class Solution:
    def HasSubtree(self, pRoot1, pRoot2):
        # write code here
        if not pRoot1 or not pRoot2:
            return False
        return self.isSame(pRoot1, pRoot2) or self.HasSubtree(pRoot1.left,pRoot2) or self.HasSubtree(pRoot1.right,pRoot2)

    def isSame(self, p,q):
        if not q:
            return True
        if p and q and p.val==q.val:
            return self.isSame(p.left,q.left) and self.isSame(p.right,q.right)
        return False

参考文献：《剑指offer》