给你两棵二叉树的根节点 p 和 q ,编写一个函数来检验这两棵树是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
提示:
两棵树上的节点数目都在范围 [0, 100] 内
-104 <= Node.val <= 104
节点定义(java版):
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
思路1:
使用深度优先遍历的思路,两棵树进行遍历,在遍历的过程中,挨个对比叶子节点。
class Solution {
public boolean isSameTree(TreeNode p, TreeNode q) {
// 判断为null的情况,两个节点都为null,则返回true,不再遍历
if(p == null && q == null){
return true;
}
// 一个节点为null,则返回false,不再遍历
if(p == null){
return false;
}
if(q == null){
return false;
}
// 如果两个节点都不为null,且两个节点的值不一致,则返回false,不再遍历
if(p.val != q.val){
return false;
}
// 递归遍历两个节点的左节点是否一致,若左节点一致,则再遍历右节点
if(isSameTree(p.left,q.left)){
return isSameTree(p.right, q.right);
}
return false;
}
}
思路2:
使用广度优先遍历的思路,两棵树进行遍历,在遍历的过程中,挨个对比叶子节点。
广度优先遍历是通过队列的先进先出特性,将树打平,挨个对比
class Solution {
public boolean isSameTree(TreeNode p, TreeNode q) {
Queue<TreeNode> queueLeft = new LinkedList();
Queue<TreeNode> queueRight = new LinkedList();
queueLeft.add(p);
queueRight.add(q);
while(!queueLeft.isEmpty() && !queueRight.isEmpty()){
TreeNode l = queueLeft.poll();
TreeNode r = queueRight.poll();
if(l == null && r != null ){
return false;
}
// 注意:两个对比的节点都为null的时候,说明这两个节点相等,还需要排查后续节点,所以是continue
if(l == null && r == null){
continue;
}
if(l != null && r == null){
return false;
}
if(l.val == r.val){
queueLeft.add(l.left);
queueLeft.add(l.right);
queueRight.add(r.left);
queueRight.add(r.right);
continue;
}
return false;
}
return queueLeft.isEmpty() && queueRight.isEmpty();
}
}