请完成一个函数,输入一个二叉树,该函数输出它的镜像。
例如输入:
4
/ \
2 7
/ \ / \
1 3 6 9
镜像输出:4
/ \
7 2
/ \ / \
9 6 3 1示例 1:
输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]
限制:
0 <= 节点个数 <= 1000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-shu-de-jing-xiang-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:利用递归,相当于后序遍历,对各个根节点的左右子节点进行调换;
//类似于两个数值进行交换,只是在二叉树的根节点的左右节点进行
class Solution {
public TreeNode mirrorTree(TreeNode root) {
if(root == null) return null;
mirrorTree(root.left);
mirrorTree(root.right);
TreeNode Temp = root.left;
root.left = root.right;
root.right = Temp;
return root;
}
}
进一步优化,在每次递归时,会有一个返回值,即已经每层已经交换好的根节点返回过来了;
class Solution {
public TreeNode mirrorTree(TreeNode root) {
if(root == null) return null;
TreeNode left = mirrorTree(root.left);
TreeNode right = mirrorTree(root.right);
root.left = right;
root.right = left;
return root;
}
}
相同嵌套的left和right必是同一层的左右叶子节点,只需将其直接赋值给root对应的叶子节点即可完成相应的交换,并且最终返回root节点到上一层嵌套。