LeetCode-检验两棵树是否相同

LeetCode-检验两棵树是否相同

题目描述

给你两棵二叉树的根节点 p 和 q ，编写一个函数来检验这两棵树是否相同。
如果两个树在结构上相同，并且节点具有相同的值，则认为它们是相同的。
例1：

例2：

例3：

思路

使用树的先序遍历，在根节点值相等的时候再分别比较左子树和右子树，要考虑两棵树结构不同的情况。

代码实现

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
def preorder(p, q):
    if not p  and  not q:
        return True
    elif not p or not q:
        return False
    elif p.val == q.val:
        a=preorder(p.left, q.left)
        b=preorder(p.right, q.right)
        return a and b 
    else:
        return False
   
class Solution(object):
    def isSameTree(self, p, q):
        """
        :type p: TreeNode
        :type q: TreeNode
        :rtype: bool
        """
        return preorder(p, q)

测试结果

结论

仍需优化。