100. Same Tree

最新推荐文章于 2024-01-17 14:26:02 发布

原创最新推荐文章于 2024-01-17 14:26:02 发布 · 266 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode

刷题专栏收录该内容

76 篇文章

订阅专栏

本文介绍了一种检查两个二叉树是否一致的方法，通过递归和迭代两种方式实现，并对比了两者的优缺点。

题目描述：

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

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

示例 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

我的思路：

直接判断p，q的值是否相等，然后递归判断p，q的left和right。或者可以层次遍历。

我的代码：

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

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

class Solution:
    def isSameTree(self, p, q):
        """
        :type p: TreeNode
        :type q: TreeNode
        :rtype: bool
        """
        if p and q:
            return p.val == q.val and self.isSameTree(p.left, q.left) and self.isSameTree(p.right, q.right)
        else:
            return p == q

Discuss：

def isSameTree2(self, p, q):
    stack = [(p, q)]
    while stack:
        node1, node2 = stack.pop()
        if not node1 and not node2:
            continue
        elif None in [node1, node2]:
            return False
        else:
            if node1.val != node2.val:
                return False
            stack.append((node1.right, node2.right))
            stack.append((node1.left, node2.left))
    return True

学到：

多考虑下问题的多种解法，能不用递归就不用吧。