判断两个树是否相等和判断tree1是否包含tree2 python实现

判断两个树是否相等

def equal(node_a, node_b):
    """
    判断两个树是否相等
    :param node_a: 
    :param node_b: 
    :return: 
    """
    if not node_a and not node_b:
        return True
    elif not node_a and node_b or node_a and not node_b or node_a.val != node_b.val:
        return False
    else:
        return equal(node_a.left, node_b.left) and equal(node_a.right, node_b.right)

判断tree1是否包含tree2

def tree1_have_tree2(tree1, tree2):
    """
    判断tree1是否包含tree2
    :param tree1: 
    :param tree2: 
    :return: 
    """
    if not tree2:
        return True
    if not tree1:
        return False
    if tree1.val != tree2.val:
        return False
    return tree1_have_tree2(tree1.left, tree2.left) and tree1_have_tree2(tree1.right, tree2.right)