题目描述:输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
两种方法:递归的和非递归的
第一种:递归的方法:
思路:
对于给定的树A和树B,如果B是A的子树,有三种情况:
第一种:A和B的根相等,这时候判断A和B的左右子树是否分别相等
第二种:B存在于A的左子树中,递归调用本函数
第三种:B存在于A的右子树中,递归调用本函数
这三种情况中,最终需要判定A和B相等,这时候才相当于在A中找到了B,需要单独的一个函数is_SameTree
如果B不是A的子树,有以下情况:
B和A至少有一个为空
所以程序编写如下:
# -*- 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
else:
return self.is_SameTree(pRoot1,pRoot2) or self.HasSubtree(pRoot1.left,pRoot2) or self.HasSubtree(pRoot1.right,pRoot2)
def is_SameTree(self,pRoot1,pRoot2):
if not pRoot2:
return True
elif not pRoot1 or pRoot1.val!=pRoot2.val:
return False
return self.is_SameTree(pRoot1.left,pRoot2.left) and self.is_SameTree(pRoot1.right,pRoot2.right)
第二种方法待更新,小伙伴们也可以自己试着实现一下。