《剑指offer》-[第3章：高质量的代码- 3.4：代码的鲁棒性] 题18：判断树的子结构

1、问题描述

输入两个二叉树A和B，判断B是不是A的子结构，如下图所示，B是A的子结构：

2、解题思路

• 边界条件：二叉树A和二叉树B的根节点都为空指针。
• 思路1：在二叉树A中查找树B时，主要分为以下两个步骤：先在树A中找到和树B根节点值一样的节点R，然后再判断以R为根节点的子树和树B是否具有相同的结构。
• 从以上可以看出，关键的操作是（1）查找和树B根节点值一样的结点R；（2）判断以R为根节点的子树是不是和B具有相同的结构。
• 接下来，我们看看（1）和（2）的具体实现过程：
• （1）查找R实际上是对树A进行遍历，可以通过递归实现，从树A的根节点开始，比较根节点的值和树B根节点是否一样，如果一样，则进行（2）操作，即判断以这两个节点为根节点的树是否具有相同的结构，如果判断结果是不具有相同结构；遍历根节点的左子树，判断以左子树为根节点的树是不是和B具有相同的结构；如果判断结果仍然是不具有相同的结构，遍历根节点的右子树，判断以右子树为根节点的树是不是和B具有相同的结构。
• （2）判断以R为根节点的子树和B是否具有相同的结构，同样可以采用递归的思路来考虑：如果节点R的值和树B根节点的值不相同，则以R为根节点的子树与树B肯定不具有相同的结构；如果相同，则递归的比较它们各自左右节点的值是否相同（具体做法：A、先判断B的左子树和R的左子树是否相同，即如果B的左子树不为空，则看R的左子树是否为空，如果为空，则直接返回不具有相同的结构，如果不为空，则递归比较R的左子树和B的左子树；B、如果左子树相同，接着判断右子树，右子树的判断和左子树类似。

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):
        # write code here
        result = False
        if pRoot1 != None and pRoot2 != None:
            if pRoot1.val == pRoot2.val:
                result = self.DoesRoot1hasRoot2(pRoot1, pRoot2)
            if result == False:
                result = self.HasSubtree(pRoot1.left, pRoot2)
            if result == False:
                result = self.HasSubtree(pRoot1.right, pRoot2)
        return result
    
    def DoesRoot1hasRoot2(self, p1, p2):
        if p2 == None:
            return True
        if p1 == None:
            return False
        if p1.val != p2.val:
        	return False
        	
        return self.DoesRoot1hasRoot2(p1.left, p2.left) and self.DoesRoot1hasRoot2(p1.right, p2.right)