剑指offer系列-面试题26-树的子结构(python)

1. 题目

输入两棵二叉树A和B，判断B是不是A的子结构。二叉树节点的定义如下：

struct BinaryTreeNode
{
	double m_dbvalue;
	BinaryTreeNode* m_pLeft;
	BinaryTreeNode* m_pRight;
}

python版

class BinaryTreeNode(object):
	def __init__(self, value, left=None, right=None):
		self.value = value
		self.left = left
		self.right = right

例子如下

---

---

---

2. 解题思路

要判断树B中的树结构是否是树A的子结构，应该如何实现呢？

1.首先判断树B的根节点是否在树A中的树中，不在直接return False；在，则进行第2步；
2.判断树A中以8为根节点的子树是不是包含和树B一样的结构。

那么我们该如何实现呢？

1.遍历树A找到与树B中根节点相同的节点；
2.判断左子节点是否相同；
3.判断右子节点是否相同；
4.递推下去，判断后续节点的左右子节点是否都相同。

3. 代码实现

3.1 递归版

def has_sub_tree(root1, root2):
	"""
	判断root2是否是root1的子结构
	"""
	result = False
	if root1 != None && root2 != None:
		if root1.value == root2.value: # 找到与树B根节点值相同的节点，判断树的结构是否相同
			result = does_tree_have_tree2(root1, root2)
		# 若树A的根节点的值不等于树B的根节点的值，则判断其左子节点是否等于树B的根节点的值
		if not result: 
			result = has_sub_tree(root1.left, root2)
		# 若树A的根节点的值不等于树B的根节点的值，则判断其右子节点是否等于树B的根节点的值
		if not result:
			result = has_sub_tree(root1.right, root2)
	return result
		
def does_tree_have_tree2(root1, root2):
	"""
	判断树的结构是否相同
	"""
	if root2 == None: # 树B为None，则树B一定是树A的子结构
		return True
	if root1 == None: # 树A为None，则树B一定不是树A的子结构，因为走到这一步说明树B一定不是None
		return False
	if root1.value != root2.value:
		return False
	return does_tree_have_tree2(root1.left, root2.left) and does_tree_have_tree2(root1.right, root2.right)
	

3.2 循环版

pass

4. 总结

树相关的算法有点复杂，不使用递归，感觉很难搞，目前先不做这方面的面试准备了。后期有需要的话在更新此博客吧。

5. 参考文献

[1] 剑指offer丛书