题目:
给你两棵二叉树的根节点 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
题解:
直接递归,首先将退出条件写好
- 处理 q 和 p 有空值的情况:
当其中一个为空另一个不为空则False,当2个都空则True
- 处理 q 和 p 都非空的情况:
当2个非空值不同则直接返回False,如果两个值相同则需要进入下一个递归
我们需要明确,只有一种情况下我们需要进行下一轮递归,就是数值相同时,我们直接将其写入return。当都有数值时我们单独考虑使用 != ,有一方没有数值或双方都没有数值的时候无法用 != 进行比较,那么我们另写一行 if(q==None or p==None): return p==q 来比较
class Solution:
def isSameTree(self, p: Optional[TreeNode], q: Optional[TreeNode]) -> bool:
if(q==None or p==None): return p==q
if(q.val != p.val): return False
return self.isSameTree(q.left,p.left) and self.isSameTree(q.right,p.right)