输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
递归思想,看每一个节点是否对应相当就好了。先判断B是不是当前节点的子树,不是再判断B是否是A的左子树的子树,然后B是否是A的左子树的子树。
中间有几个注意的,如果A遍历完了,而B没完,说明B比A大,不是子树;如果A完了,B完了;如果A没完,B完了,同时所有节点都能对上,就说明B是A的子树。返回True
贴上代码:
# -*- 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
return self.compar(pRoot1,pRoot2) or self.compar(pRoot1.left,pRoot2) or self.compar(pRoot1.right,pRoot2)
def compar(self,node1,node2):
if(node2==None):
return True
if(node1==None):
return False
if(node1.val!=node2.val):
return False
return self.compar(node1.left,node2.left) and self.compar(node1.right,node2.right)
祝大家offer多多!