给你两棵二叉树的根节点 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
1.自己的解法:深度优先算法。逐个判断每个节点是否相等,如果当前节点不相等,返回false,如果当前节点相等,通过递归继续判断左右子节点是否相等。
/**
* 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;
* }
* }
*/
class Solution {
public boolean isSameTree(TreeNode p, TreeNode q) {
if(p == null && q == null){//如果两棵树都为空,则返回true
return true;
}
if((p == null && q != null) || (p != null && q == null)){//如果其中一棵为空,另一棵不为空,则返回false
return false;
}
if(p.val == q.val){//如果两棵树的当前节点的值相等
return isSameTree(p.left,q.left) && isSameTree(p.right,q.right);//继续向下分别判断左右两个子节点
}else{//当前节点的值不相等,返回false
return false;
}
}
}
2.答案提供的另一种思路:广度优先算法。用两个队列来存储遍历到的节点,先判断当前节点的子结构是否一致,不一致就返回true,如果一致就把存在的节点加入队列,然后进行值的比较。
class Solution {
public boolean isSameTree(TreeNode p, TreeNode q) {
if (p == null && q == null) {
return true;
} else if (p == null || q == null) {
return false;
}
//用两个队列存储遍历到的节点
Queue<TreeNode> queue1 = new LinkedList<TreeNode>();
Queue<TreeNode> queue2 = new LinkedList<TreeNode>();
queue1.offer(p);//将两个根节点加入队列
queue2.offer(q);
while (!queue1.isEmpty() && !queue2.isEmpty()) {//两个队列不为空的情况下,即还有节点未被比较
TreeNode node1 = queue1.poll();//取出队列的头节点进行比较
TreeNode node2 = queue2.poll();
if (node1.val != node2.val) {//如果值不相等,返回false
return false;
}
//如果值相等,继续比较左右子结点
TreeNode left1 = node1.left, right1 = node1.right, left2 = node2.left, right2 = node2.right;
if (left1 == null ^ left2 == null) {//两棵树中只有一棵有左子节点时
return false;
}
if (right1 == null ^ right2 == null) {//两棵树中只有一棵有右子节点时
return false;
}
//将存在的左右子节点存入队列,进行后续的比较
if (left1 != null) {
queue1.offer(left1);
}
if (right1 != null) {
queue1.offer(right1);
}
if (left2 != null) {
queue2.offer(left2);
}
if (right2 != null) {
queue2.offer(right2);
}
}
return queue1.isEmpty() && queue2.isEmpty();//只有当两个队列均为空时,即均比较完毕,才返回true。
}
}
题源:力扣