python - 如何使用顺序遍历比较树

Question

问题要求使用顺序遍历检查树q 和p 是否相同。我的逻辑是：

1. 在检查 p 和 q 的左右时，如果 q 有左/右孩子但 p 没有，则返回 False

2. 检查p和q的值时，如果不相同则返回False

我在示例中找到：（BFS 订单）

p = [1,2,3,5,6]

q = [1,2,3,4,6]

它发现了差异，但仍然返回 True。我有点理解这是因为它只在递归中调用inOrderTraverse()，所以return 命令不会停止递归运行。我尝试在递归部分之前添加return，使其变为return inOrderTraverse()。但是，在这种情况下，一旦访问了所有剩余的孩子，递归就会停止。

所以我真的不知道我还能做些什么改变来让这件事发挥作用。请帮忙，谢谢！

def inOrderTraverse(p, q):

     if p.left:
         if q.left and p.left.val == q.left.val:
             inOrderTraverse(p.left, q.left)
         else:
             return False

     if p.val != q.val:
         return False

     if p.right:
         if q.right and p.right.val == q.right.val:
             inOrderTraverse(p.right, q.right)
         else:
             return False

     return True

Answer 1

除了忽略对inOrderTraverse() 的递归调用的返回值，如 cmets 中所述，您似乎错过了q.left 存在但p.left 不存在的情况（同上）。这是我为解决这两个问题而对您的代码进行的修改：

def inOrderTraverse(p, q):

    if p.left or q.left:
        if p.left and q.left:
            if not inOrderTraverse(p.left, q.left):
                return False  # sub trees don't match
        else:
            return False  # only one has left

    if p.val != q.val:
        return False  # vals don't match

    if p.right or q.right:
        if p.right and q.right:
            if not inOrderTraverse(p.right, q.right):
                return False  # sub trees don't match
        else:
            return False  # only one has right

    return True  # everything matches

Answer 2

除了返回递归值，逻辑还可以简化一点：

def areEqual(p, q):

  return bool(p) == bool(q) and (p is None or areEqual(p.left, q.left) and p.val == q.val and areEqual(p.right, q.right))