给定两个二叉树,编写一个函数来检验它们是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
示例 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的根节点是否相同,然后遍历左右节点。代码如下:
# 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
"""
# 1.首先判断p,q为空的情况
if p==None and q==None:
return True
elif p==None and q!=None:
return False
elif p!=None and q==None:
return False
# 2.都不为空时,先判断根节点的值是否相同
if p.val!=q.val:
return False
l = self.isSameTree(p.left, q.left)#再判断左子树是否相同,返回的值用l接受
r = self.isSameTree(p.right, q.right)#最后判断右子树是否相同,返回的值用r接受
# 注意:这种情况下,python使用函数递归时,是self.函数名(参数),如果原函数参数没有self,那递归调用和c语言一样
return l and r