题目描述:
输入一个二叉树,输出它的镜像。
递归法:
根据二叉树镜像的定义,考虑递归遍历二叉树,交换每个结点的左右子节点,即可生成二叉树的镜像。
递归解析:
- 终止条件:当结点root为空时(即越过叶节点),则返回Null
- 递推工作:
- 初始化结点tmp,用于暂存root的左子节点
- 开启递归右子节点mirrorTree(root.right),并将返回值作为root的左子节点
- 开启递归左子节点mirrorTree(tmp),并将返回值作为root的右子节点
- 返回值:
- 返回当前结点root
public TreeNode mirrorTree(TreeNode root){
if(root == null)
return null;
TreeNode tmp = root.left;
root.left = mirrorTree(root.right);
root.right = mirrorTree(tmp);
return root;
}
复杂度分析:
- 时间复杂度:O(n),其中n为二叉树的结点数量,建立二叉树镜像需要遍历树的所有结点,占用O(n)时间
- 空间复杂度:O(n),最差情况下,递归时系统需要使用O(n)大小的栈空间
辅助栈(或队列):
利用栈(或队列)遍历树的所有结点node,并交换每个node的左右子节点
算法流程:
- 特例处理:当root为空时,直接返回null
- 初始化:栈(或队列),并加入根节点root
- 循环交换:当栈stack为空时跳出:
- 出栈:记为node
- 添加子节点:将node左右子节点入栈
- 交换:交换node的左右 子节点
- 返回值:返回根节点root
算法java实现:
public TreeNode mirrorTree(TreeNode root){
if(root == null)
return null;
Stack<TreeNode> stack = new Stack<>() {{ add(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 tmp = node.left;
node.left = node.right;
node.right = tmp;
}
}
复杂度分析:
- 时间复杂度:O(n),其中n为结点数量,建立二叉树镜像需要遍历所有结点,占用O(n)时间
- 空间复杂度:O(n),最差情况下,栈stack最多同时存储n/2个结点,占用O(n)额外空间