【牛客网】树的子结构解题报告 (python)

最新推荐文章于 2021-07-23 15:08:30 发布

暮雨凉初透

最新推荐文章于 2021-07-23 15:08:30 发布

阅读量194

点赞数

分类专栏：树

本文链接：https://blog.csdn.net/qq_32805671/article/details/104220731

版权

树专栏收录该内容

5 篇文章 0 订阅

订阅专栏

原题地址：https://www.nowcoder.com/practice/6e196c44c7004d15b1610b9afca8bd88?tpId=13&tqId=11170&tPage=1&rp=1&ru=%2Fta%2Fcoding-interviews&qru=%2Fta%2Fcoding-interviews%2Fquestion-ranking

题目描述：

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

解题方案：

采用递归方法，原函数不太好递归，重写递归函数。

# -*- 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 and pRoot2):
            return False
        return self.is_subtree(pRoot1, pRoot2) or self.HasSubtree(pRoot1.left, pRoot2) or self.HasSubtree(pRoot1.right, pRoot2)
     
    def is_subtree(self, A, B):
        if not B:
            return True
        if not A or A.val != B.val:
            return False
        return self.is_subtree(A.left, B.left) and self.is_subtree(A.right, B.right)