给你一个二叉树的根节点 root ,检查它是否轴对称。
示例1:
输入:root = [1,2,2,3,4,4,3]
输出:true
1
/ \
2 2
/ \ / \
3 4 4 3
示例2:
输入:root = [1,2,2,null,3,null,3]
输出:false
1
/ \
2 2
\ \
3 3
分析
对于根节点来说,左子树和右子树每个节点都一致,则表示二叉树轴对称。
那么需要有个方法可以对比左子树的每一个节点和右子树的每一个节点
private boolean check(TreeNode left, TreeNode right) {
}
两个节点自身比较是比较简单的,只要比较节点的值是否相等
left.val == right.val
但是如何递归比较呢?
参考示例1的二叉树:
1
/ \
2 2
/ \ / \
3 4 4 3
根节点的左子树的左子节点 3 需要和 根节点的右子树的右子节点 3 比较
根节点的左子树的右子节点 4 需要和 根节点的右子树的左子节点 4 比较
即应该递归比较 left 节点的左子节点和 right 节点的右子节点并且同时比较 left 节点的右子节点和 right 节点的左子节点
left.val == right.val && check(left.left, right.right) && check(left.right, right.left)
边界情况分析
- 如果 left 和 right 都为空,那么可以认为递归比较结束了,无需继续递归调用,且之前的节点都是相同的,二叉树是轴对称的
- 如果 left 或 right 有一个为空,那么二叉树肯定就不轴对称了
完整的 check 方法:if(left == null && right == null) {
return true;
}
if(left == null || right == null) {
return false;
}
但是 isSymmetric 方法如何调用 check 方法呢?private boolean check(TreeNode left, TreeNode right) {
if(left == null && right == null) {
return true;
}
if(left == null || right == null) {
return false;
}
return left.val == right.val && check(left.left, right.right) && check(left.right, right.left);
}
我们可以把示例1 理解为如下的两棵树:
1 1
/ \
2 2
/ \ / \
3 4 4 3
那么 isSymmetric 方法调用可以按如下方式调用 check 方法:
check(root, root);
完整递归算法
public boolean isSymmetric(TreeNode root) {
return check(root, root);
}
private boolean check(TreeNode left, TreeNode right) {
if(left == null && right == null) {
return true;
}
if(left == null || right == null) {
return false;
}
return left.val == right.val && check(left.left, right.right) && check(left.right, right.left);
}
递归如何转迭代?
递归都是可以转迭代的,递归转迭代需要借用其他的数据结构暂存一下需要比较的元素(比如队列或者栈),并且需要一直循环比较,直至所有的元素都完成比较:
private boolean check(TreeNode left, TreeNode right) {
Queue<TreeNode> queue = new LinkedList<TreeNode>();
while(!queue.isEmpty()) {
}
}
那在 while 循环前需要做什么?在 while 循环里面又需要做什么?什么时候跳出结束 while 循环?
- while 循环前需要提前往队列里面放入需要比较的元素;
while 循环里面就是比较两个节点是否一致以及是否需要结束循环
- 从队列中取出需要比较的元素,并比较元素是否一致
- 比较时如果检查发现树已经不满足轴对称了,则可以提前跳出结束 while 循环了
一次循环结束,在下次循环开始之前,还需要重新往队列中放入需要比较的元素:private boolean check(TreeNode left, TreeNode right) {
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(left);
queue.offer(right);
while(!queue.isEmpty()) {
right = queue.poll();
left = queue.poll();
if(left == null && right == null) {
continue;
}
if((left == null || right == null) || (left.val != right.val)) {
return false;
}
}
}
参考递归中的分析:比较 left 节点的左子节点和 right 节点的右子节点并且同时比较 left 节点的右子节点和 right 节点的左子节点
所以往队列中放入元素的顺序有要求:
放入 left 节点的左子节点后需要放入 right 节点的右子节点
放入 left 节点的右子节点后需要放入 right 节点的左子节点
private boolean check(TreeNode left, TreeNode right) {
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(left);
queue.offer(right);
while(!queue.isEmpty()) {
right = queue.poll();
left = queue.poll();
if(left == null && right == null) {
continue;
}
if((left == null || right == null) || (left.val != right.val)) {
return false;
}
queue.offer(left.left);
queue.offer(right.right);
queue.offer(left.right);
queue.offer(right.left);
}
}
完整迭代算法
```java public boolean isSymmetric(TreeNode root) { return check(root, root); }
private boolean check(TreeNode left, TreeNode right) {
Queue
queue.offer(left.left);
queue.offer(right.right);
queue.offer(left.right);
queue.offer(right.left);
}
return true;
} ```