给你两棵二叉树的根节点 p 和 q ,编写一个函数来检验这两棵树是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
示例 1:
输入:p = [1,2,3], q = [1,2,3]
输出:true
示例 2:
输入:p = [1,2], q = [1,null,2]
输出:false
示例 3:
输入:p = [1,2,1], q = [1,1,2]
输出:false
提示:
两棵树上的节点数目都在范围 [0, 100] 内
-104 <= Node.val <= 104
思路:
对两棵树进行遍历(采用递归实现)
边遍历边比对,出现不同的节点则返回false
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
def traversal (root1, root2):
if root1 == None and root2 != None:
return False
elif root1 != None and root2 == None:
return False
elif root1 == None and root2 == None:
return True
else:
return root1.val == root2.val and traversal(root1.left, root2.left) and traversal(root1.right, root2.right)
return traversal(p, q)