100. 相同的树
给你两棵二叉树的根节点 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
解题思路
深度优先搜索:
如果两个二叉树都为空,则两个二叉树相同。如果两个二叉树中有且只有一个为空,则两个二叉树一定不相同。如果两个二叉树都不为空,则先判断它们的根节点的值是否相同,若不相同则两个二叉树一定不同,若相同,再利用判断两个二叉树的左、右子树是否相同。这是一个递归的过程,因此可以使用深度优先搜索,递归地判断两个二叉树是否相同。
C语言
bool isSameTree(struct TreeNode* p, struct TreeNode* q) {
if (p == NULL && q == NULL) {//两个节点都为空时,返回true
return true;
} else if (p == NULL || q == NULL) {//若某个节点为空,另一个节点不为空,则返回false
return false;
} else if (p->val != q->val) {//若两个节点都不为空,但是两个节点的值不相等,则返回false
return false;
} else {//若两个节点都不为空,且两个节点的值相等,则利用递归,判断它们的左右子树是否相等
return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
}
}
Java
class Solution {
public boolean isSameTree(TreeNode p, TreeNode q) {
if (p == null && q == null) {//两个节点都为空时,返回true
return true;
} else if (p == null || q == null) {//若某个节点为空,另一个节点不为空,则返回false
return false;
} else if (p.val != q.val) {//若两个节点都不为空,但是两个节点的值不相等,则返回false
return false;
} else {//若两个节点都不为空,且两个节点的值相等,则利用递归,判断它们的左右子树是否相等
return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
}
}
}
Python
class Solution(object):
def isSameTree(self, p, q):
"""
:type p: TreeNode
:type q: TreeNode
:rtype: bool
"""
if not p and not q: # 两个节点都为空时,返回true
return True
elif not p or not q: # 若某个节点为空,另一个节点不为空,则返回false
return False
elif p.val != q.val: # 若两个节点都不为空,但是两个节点的值不相等,则返回false
return False
else: # 若两个节点都不为空,且两个节点的值相等,则利用递归,判断它们的左右子树是否相等
return self.isSameTree(p.left,q.left) and self.isSameTree(p.right,q.right)