给定两个二叉树,编写一个函数来检验它们是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
示例 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
思路:
1.大多数的二叉树题目都是用递归可以解的。
所以当拿到二叉树的题目的时候,我们首先就是看看能拆解成哪些子问题。
这个问题的子问题很简单,就是左子树,右子树都相等的二叉树是相同的二叉树。注意出口条件。
// 递归
public boolean isSameTree(TreeNode p, TreeNode q) {
if (p == null && q == null) { // 都为空
return true;
}
if (p == null || q == null) { // 由于是从上判断下来的 不可能都为空
return false; // 只有一个为空则false
}
if (p.val != q.val) {
return false; // 值不等
}
return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
}
2.用非递归解怎么解呢?
如果遇到二叉树的问题,没思路还有第二招,就是想想看是不是遍历的变种:
-
先序遍历
-
中序遍历
-
后序遍历
-
层次遍历
我们可以用队列,一起进行层序遍历,同时比较左右两颗树:
// 非递归 两棵树层序比较
public boolean isSameTree2(TreeNode p, TreeNode q) {
if (p == null && q == null) { // 都为空
return true;
}
if (p == null || q == null) { // 由于是从上判断下来的 不可能都为空
return false; // 只有一个为空则false
}
Queue<TreeNode> queue = new LinkedList<>();
queue.add(p);
queue.add(q);
while (!queue.isEmpty()) {
TreeNode first = queue.poll();
TreeNode second = queue.poll();
if (first == null && second == null) {
// 都是空的,遍历到叶子节点了
continue;
} else if (first == null || second == null) {
// 有一个为null
return false;
} else if (first.val != second.val) {
// 不相等.
return false;
}
// 左子树入队
queue.add(first.left);
queue.add(second.left);
// 右子树入队
queue.add(first.right);
queue.add(second.right);
}
return true;
}