请完成一个函数,输入一个二叉树,该函数输出它的镜像。
思路:递归或者栈。递归交换每个节点的左右节点,实现得到二叉树镜像。注意要先暂存树的左子节点,如果不暂存的话递归完右子节点后左子节点的值就改变了。
/**
* 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;
}
TreeNode temp = root.left; //暂存左子节点
root.left = mirrorTree(root.right);//递归右子节点
root.right = mirrorTree(temp);//递归左子节点
return root;
}
}
//栈
class Solution {
public TreeNode mirrorTree(TreeNode root) {
if (root == null){
return null;
}
Stack<TreeNode> Stack = new Stack<>();
Stack.push(root);//根节点入栈
while(!Stack.isEmpty()){
TreeNode node = Stack.pop();//节点出栈,出栈节点左右子节点入栈,并交换位置
if(node.left != null) Stack.add(node.left);
if(node.right != null) Stack.add(node.right);
TreeNode temp = node.left;
node.left = node.right;
node.right = temp;
}
return root;
}
}