首先,我们需要实现一个函数来比较两个二叉树是否相等。这可以通过递归来实现。对于每个节点,我们检查它们的值是否相等,并且递归地比较它们的左子树和右子树,如果它们都相等,返回true,否则返回false。
接下来,我们可以通过递归遍历搜索树来找到给定子树。对于每个节点,我们检查它们的值是否相等,并且递归地搜索它们的左子树和右子树,如果找到匹配的子树,返回true,否则返回false。
这是一个Python示例代码:
class TreeNode: def init(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right
def is_same_tree(p: TreeNode, q: TreeNode) -> bool: if p is None and q is None: return True elif p is None or q is None: return False elif p.val != q.val: return False else: return is_same_tree(p.left, q.left) and is_same_tree(p.right, q.right)
def is_subtree(s: TreeNode, t: TreeNode) -> bool: if s is None: return False elif is_same_tree(s, t): return True else: return is_subtree(s.left, t) or is_subtree(s.right, t)
在这里,我们首先实现了is_same_tree函数,然后在is_subtree函数中使用它来查找子树。
上一篇:比较两个二叉搜索树