判断相同的二叉树

LeetCode：判断相同的二叉树

 

给定两个二叉树，编写一个函数来检验它们是否相同。

如果两个树在结构上相同，并且节点具有相同的值，则认为它们是相同的。

示例 1:

输入:       1         1
          / \       / \
         2   3     2   3

        [1,2,3],   [1,2,3]

输出: true
示例 2:

输入:      1          1
          /           \
         2             2

        [1,2],     [1,null,2]

输出: false
示例 3:

输入:       1         1
          / \       / \
         2   1     1   2

        [1,2,1],   [1,1,2]

输出: false

思路：同时遍历2棵二叉树，发现不同便终止，否则继续

定义二叉树：

# Definition for a binary tree node.
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

递归实现：

    # 递归
    def isSameTree2(self, p: TreeNode, q: TreeNode) -> bool:
        if not p and not q: return True
        if p and q and p.val==q.val:
            return self.isSameTree(p.left,q.left) and self.isSameTree(p.rihgt,q.right)

        

迭代实现：

class Solution:
    def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
        stack=[(p,q)]
        while stack:
            a,b = stack.pop()
            if not a and not b:
                continue
            if a and b and a.val==b.val:
                stack.append((a.right,b.right))
                stack.append((a.left,b.left))
            else:
                return False
        return True