二叉树里的双指针
所谓的双指针就是定义了两个变量,在二叉树中有时候需要用至少两个变量才能解决问题,这两个指针可能是针对一棵树,也可能是两棵树,一般与对称、反转和合并相关。
1、判断两颗树是否相同
LeetCode100. 给你两颗二叉树的节点p和q,编写一个函数来检验这两颗树是否相同。如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
示例1:
输入:p = [1, 2, 3], q = [1, 2, 3]
输出:true
示例2:
输入:p = [1, 2], q = [1, null, 2]
输出:false
对于这个题,只需要对两个二叉树进行前序遍历,先判断根节点是否相同,如果相同再分别判断左右子节点是否相同,判断的过程中只要有一个不相同就返回false,如果全部相同才会返回true。
代码如下所示:
public boolean isSameTree(TreeNode p, TreeNode q) {
//如果都为空就认为是相同的
if (p == null && q == null) {
return true;
}
//如果一个为空,一个不为空,则是不相同的树,返回false
if (p == null || q == null) {
return false;
}
//如果对应位置的两个节点的值不相等,则也不是用一棵树
if (p.val != q.val) {
return false;
}
//走到这一步说明节点p和q是完全相同的,接下来再判断其左子树和右子树是否相同即可
return isSameTree(p.left, q.left) && isSameTree(p.right, q.right)
}
2、判断两棵树是否对称
LeetCode101. 给定一个二叉树,检查它是否是镜像对称的。
示例1:
输入:[1, 2, 2, null, 3, null, 3]
输出:false
示例2:
输入:[1, 2, 2, 3, 4, 3, 4]
输出:true
因为我们要通过递归函数的返回值来判断两个子树的内部节点和外部节点是否相同,所以准确的说一个树的遍历顺序是左右中,一个树的遍历顺序是右左中。这里的关键是如何比较和如何处理结束的条件。单层递归的逻辑就算处理左右节点都不为空,且数值相同的情况
1.比较二叉树外部是否对称:传入的是左节点的左孩子,右节点的右孩子
2.比较内部是否对称,传入左节点的右孩子,右节点的左孩子
3.如果左右都对称就返回true,有一侧不对称就返回false
具体的代码如下:
class Solution {
//主方法
public boolean isSymmetric (TreeNode root) {
if (root == null){
return true;
}
return check(root.left, root.right);
}
public boolean check(TreeNode p, TreeNode q) {
if (p == null && q == null) {
return true;
}
if (p == null && q == null) {
return false;
}
if (p.val != q.val) {
return false;
}
return check(p.left, q.right) && check(p.right, q.left);
}
}
3、合并二叉树
LeetCode617. 给定两个二叉树,想象当你将它们中的一个覆盖到另一个上是,两个二叉树的一些节点便会重叠。你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为null的节点将直接作为新二叉树的节点。
两个二叉树的对应节点可能存在以下三种情况,对于每种情况使用不同的合并方式。
1.如果两个二叉树的对应节点都为空,则合并后的二叉树的对应节点也为空
2.如果两个二叉树的对应节点只有一个为空,则合并后的二叉树的对应节点为其中的非空节点
3.如果两个二叉树的对应节点都不为空,则合并后的二叉树的对应节点的值为两个二叉树的对应节点的值之和,此时需要显性合并两个节点
4.对一个节点进行合并后,还要对该节点的左右子树分别进行合并
具体的代码实现如下:
public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
if (t1 == null){
return t2;
}
if (t2 == null) {
return t1;
}
TreeNode merged = new TreeNode(t1.val + t2.val);
merged.left = mergeTrees(t1.left, t2.left);
merged.right = mergeTrees(t1.right, t2.right);
return merged;
}