请完成一个函数,输入一个二叉树,该函数输出它的镜像。
示例:
输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-shu-de-jing-xiang-lcof
解题思路
广度遍历二叉树, 把每个节点的左右子节点交换
广度遍历:用一个队列或栈, 现将根节点加入队列, 然后只要队列不为空就循环执行如下操作
- 将队首弹出, 判断左右子树是否为空, 不为空则加入队列
- 然后交换左右子树
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode mirrorTree(TreeNode root) {
if (root == null) {
return null;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
TreeNode tmp;
while (!queue.isEmpty()) {
TreeNode cur = queue.remove();
if (cur.left != null) {
queue.add(cur.left);
}
if (cur.right != null) {
queue.add(cur.right);
}
tmp = cur.right;
cur.right = cur.left;
cur.left = tmp;
}
return root;
}
}