题目描述
操作给定的二叉树,将其变换为源二叉树的镜像。
输入描述:
二叉树的镜像定义:源二叉树
8
/ \
6 10
/ \ / \
5 7 9 11
镜像二叉树
8
/ \
10 6
/ \ / \
11 9 7 5
思路:
使用递归,先到最低层,然后一层层往上交换。
注意为null的元素要直接return
public class Solution {
public void Mirror(TreeNode root) {
if(root==null)
return ;
if(root.left!=null||root.right!=null){
Mirror(root.left);
Mirror(root.right);
}
// 如果root的左右节点都是null,也会进入这个方法,会增加时间复杂度,不好
TreeNode temp = root.left;
root.left = root.right;
root.right = temp;
}
}
也可以先交换,再递归
public void Mirror(TreeNode root) {
if(root==null)
return;
if(root.left==null&&root.right==null)
return;
TreeNode temp = root.left;
root.left = root.right;
root.right = temp;
if(root.left!=null)
Mirror(root.left);
if(root.right!=null)
Mirror(root.right);
}
或者不用递归做,用栈来存储节点(整体思路类似前序遍历)
import java.util.*;
public class Solution {
Stack<TreeNode> stack = new Stack<>();
public void Mirror(TreeNode root) {
if(root==null)
return ;
stack.push(root);
while(!stack.empty()){
root = stack.pop();
TreeNode temp = root.left;
root.left = root.right;
root.right = temp;
if(root.right!=null)
stack.push(root.right);
if(root.left!=null)
stack.push(root.left);
}
}
}