剑指offer-树的子结构-python

题目描述

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

思路

1. 空树不是任意一个树的子结构，如果 root1 与root2 中有一个为空树的话，返回False
2. 判断是否存在子树：
   1. 1. 当rootA的值与rootB一致时，B是A的子结构（且A的左子树等于B左子树，A的右子树等于B右子树）
      2. 当A的左子树等于B，B是A的子结构
      3. 当A的右子树等于B，B是A的子结构
3. 循环递归

# -*- 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):

        if not pRoot1 or not pRoot2:

            return False

        return self.isrubtreeequl(pRoot1,pRoot2) or self.HasSubtree(pRoot1.left, pRoot2) or self.HasSubtree(pRoot1.right, pRoot2)

    def isrubtreeequl(self,root1,root2):

        if not root2:

            return True

        if not root1:

            return False

        return root1.val == root2.val and self.isrubtreeequl(root1.left,root2.left) and self.isrubtreeequl(root1.right,root2.right)