剑指offer树的子结构

题目描述

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

思路

     这道题我们要用递归去做，为了满足空树不是任意一个树的子结构，我们还需再定义另外一个函数来实现递归判断（如果一旦把这个条件放到递归，一遍历完树2，就返回False，这明显是错的）

1. 如果树2被遍历完了，返回True
2. 如果树1被遍历完了，让树1和树2的当前结点比较是否相同
3. 树1和树2的值相等，就接着遍历其左右子节点
4. 不相等就分别遍历树1的左，右结点和树2的当前结点比较

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def HasSubtree(self, pRoot1, pRoot2):
        # write code here
        if not pRoot1 or not pRoot2:
            return False
        return self.judge(pRoot1,pRoot2)
    def judge(self,p1,p2):
        if not p2:
            return True
        if not p1:
            return p1==p2
        res=False
        if p1.val==p2.val:
            res=self.judge(p1.left,p2.left) and self.judge(p1.right,p2.right)
        return res or self.judge(p1.left,p2) or self.judge(p1.right,p2)

测试用例

if __name__=='__main__':
    s=Solution()
    #构建树1
    root1=TreeNode(2)
    node1=TreeNode(2)
    node2=TreeNode(2)
    node3=TreeNode(4)
    node4=TreeNode(1)
    node5=TreeNode(3)
    root1.left=node1
    node1.left=node2
    node1.right=node3
    node2.left=node4
    node2.right=node5
    root1.right=node4.left=node4.right=node5.left=node5.right=node3.left=node3.right=None
    #构建树2
    root2=TreeNode(2)
    node5=TreeNode(1)
    node6=TreeNode(3)
    root2.left=node5
    root2.right=node6
    node5.left=node5.right=node6.left=node6.right=None
    #调用函数
    print(s.HasSubtree(root1,root2))