《剑指offer》之树的子结构（python实现）

 

问题描述：

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

思路分析：
（思路整体上参考了牛客网评论区的答案）
对于两棵二叉树来说，要判断B是不是A的子结构----首先第一步在树A中查找与B根节点的值一样的节点；
第二步就是判断树A中以R为根节点的子树是不是和树B具有相同的结构；
如果节点R的值和树的根节点不相同，则以R为根节点的子树和树B肯定不具有相同的节点，也就是B不是A的子树；
如果它们值是相同的，则递归的继续判断各自的左右节点的值是不是相同，终止条件是我们达到了树A或者树B的叶节点。

 

python实现代码如下：

# -*- 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.DoesP1HaveP2(pRoot1,pRoot2)
            if not result:
                result = self.DoesP1HaveP2(pRoot1.left,pRoot2)
            if not result:
                result = self.DoesP1HaveP2(pRoot1.right,pRoot2)
            return result
    def DoesP1HaveP2(self, ptr1, ptr2):
        if ptr2 == None:
            return True
        if ptr1 == None:
            return False
        if ptr1.val != ptr2.val:
            return False
        return self.DoesP1HaveP2(ptr1.left, ptr2.left) and self.DoesP1HaveP2(ptr1.right, ptr2.right)

另一种方式代码如下：

（看到评论区一位华科平凡的大神将其转换为字符串来判断是不是含有相同的子结构，nb啊膜拜各路大神，默默学习）

class Solution:
    def HasSubtree(self, pRoot1, pRoot2):
         
        def convert(p):
            if p:
                return str(p.val) +  convert(p.left) + convert(p.right)
            else:
                return ""
        return convert(pRoot2) in convert(pRoot1) if pRoot2 else False