LeetCode *100. Same Tree (Python Solution)

最新推荐文章于 2022-03-09 22:42:35 发布

原创最新推荐文章于 2022-03-09 22:42:35 发布 · 265 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode题解专栏收录该内容

16 篇文章

订阅专栏

该博客主要探讨LeetCode中的第100题，即检查两个二叉树是否在结构和节点值上相同。提供了两种Python解决方案：递归法（solution 1.1和优化后的solution 1.2）。递归法虽然易于理解，但时间复杂度较高。

题目描述

Given two binary trees, write a function to check if they are the same or not.

Two binary trees are considered the same if they are structurally identical and the nodes have the same value.

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

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

Example 1:

Input:     1         1
          / \       / \
         2   3     2   3

        [1,2,3],   [1,2,3]
        
Output: true

Example 2:

Input:     1         1
          /           \
         2             2

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

Output: false

Example 3:

Input:     1         1
          / \       / \
         2   1     1   2

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

Output: false

Python Solution

分析： 这类问题大致都有两类方法。solution 1 为 递归法 ，重复地调用自己，优点代码易读好理解，缺点复杂度很高。Solution 2 为 迭代法 ，

递归法代码如下：

solution 1.1

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
        if not p and not q:
            return True
        if not p or not q:
            return False
        if p.val != q.val:
            return False
        return self.isSameTree(p.left, q.left) and self.isSameTree(p.right, q.right)

这段代码进行了3次判定，第一次为 p 、 q 是否同时为空，是则相等，第二次为其中有一次为空则不等，第三次就是都不为空的时候判断 p 、q 的值是否相等。

对 solution 1.1 代码进行优化：

solution 1.2

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
        if p and q:
            return p.val == q.val and self.isSameTree(p.left, q.left) and self.isSameTree(p.right, q.right)
        return p is q