剑指offer-17.树的子结构-Python

17.树的子结构

题目描述

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

记录

该实例可以很好说明多种情况：

(1) B可能是从A的根节点出发的子树
(2) B可能是从A的左子节点出发的子树
(3) B可能是从A的右子节点出发的子树

(1)
虽然A和B的根节点值相同，但是子节点值不相同
self.IsSubTree(pRoot1, pRoot2)：True
self.IsSubTree(pRoot1.left, pRoot2.left) and self.IsSubTree(pRoot1.right, pRoot2.right)：False
(2)
self.IsSubTree(pRoot1.left, pRoot2)：True
self.IsSubTree(pRoot1.left, pRoot2.left) and self.IsSubTree(pRoot1.right, pRoot2.right)：True
if pRoot2 is None: return True（先判断B的重要性就体现出来了，如果先判断A是否走完就会为False，则出错。可以看代码注释）
(3)
self.IsSubTree(pRoot1.right, pRoot2)：False

结果：True

# -*- 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.IsSubTree(pRoot1, pRoot2) or self.IsSubTree(pRoot1.left, pRoot2) or self.IsSubTree(pRoot1.right, pRoot2)
        
    def IsSubTree(self, pRoot1, pRoot2):
    	#B的节点遍历结束，且B所有的节点都可以对应上
        if pRoot2 is None:
            return True
        #B没遍历完，A却遍历结束，B肯定不是A的子树
        if pRoot1 is None:
            return False
        #确保每个节点都可以对应
        if pRoot1.val != pRoot2.val:
            return False
        return self.IsSubTree(pRoot1.left, pRoot2.left) and self.IsSubTree(pRoot1.right, pRoot2.right)