剑指offer32 --- 树的子结构（Python）

剑指offer32 — 树的子结构（Python）

题目

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

分析

先判断root的情况（判断root的情况时也是先判断root，再判断左右子树是否一致（递归））, 再判断左右子树（递归）。具体见代码。

解

# -*- 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 pRoot1==None or pRoot2==None:
            return False
        
        def HasEqual(pRoot1,pRoot2):
            if pRoot2 == None:
                return True
            if pRoot1 == None:
                return False
            
            if pRoot1.val == pRoot2.val:
                if pRoot2.left == None:
                    leftEqual = True
                else:
                    leftEqual = HasEqual(pRoot1.left,pRoot2.left)
                    
                if pRoot2.right == None:
                    rightEqual = True
                else:
                    rightEqual = HasEqual(pRoot1.right,pRoot2.right)
                return leftEqual and rightEqual #左右均相等的时候才返回True
            return False
        
        # 1. 判断根结点的情况
        if pRoot1.val == pRoot2.val:
            ret = HasEqual(pRoot1,pRoot2) #递归判断是否相等
            if ret:
                return True
            
        # 2. 判断左子树的情况
        ret = self.HasSubtree(pRoot1.left, pRoot2)       
        if ret:
            return True
        
        # 3. 判断右子树的情况
        ret = self.HasSubtree(pRoot1.right, pRoot2)
        if ret:
            return True

牛客网该题链接