码迷,mamicode.com
首页 > 编程语言 > 详细

剑指offer树的子结构python

时间:2019-11-04 11:27:38      阅读:98      评论:0      收藏:0      [点我收藏+]

标签:desc   describe   root   __init__   offer   遍历   content   item   二叉树   

题目描述

输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)

思路

首先遍历a树,找和b树根结点值相等的结点,找到后判断是不是子树。

判断是不是子树的时候,用递归的方法,判断根节点,再判断左子右子。

代码

# -*- 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 and pRoot2:
            if pRoot1.val == pRoot2.val:
                result = self.isSubtree(pRoot1,pRoot2)
            if not result and pRoot1.left:
                result = self.isSubtree(pRoot1.left,pRoot2)
            if not result and pRoot2.right:
                result = self.isSubtree(pRoot1.right,pRoot2)
        return result
    
    def isSubtree(self,p1,p2):
        if not p2:
            return True
        if not p1:
            return False
        if p1.val != p2.val:
            return False
        check_left = self.isSubtree(p1.left, p2.left)
        check_right = self.isSubtree(p1.right,p2.right)
        return check_left and check_right
    

 

剑指offer树的子结构python

标签:desc   describe   root   __init__   offer   遍历   content   item   二叉树   

原文地址:https://www.cnblogs.com/wangzhihang/p/11790889.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!