一、题目描述
请完成一个函数,输入一个二叉树,该函数输出它的镜像。
例如输入:
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
二、思路讲解
1、方法一:递归(dfs)
和树有关的题目一般都离不开深度优先遍历。
我们将根节点的左右子树调换,然后将他的左子树的左右子树调换、将它的右子树的左右子树调换……这就是递归的思路了。
然后是跳出递归的条件:如果给的树为空,或者是已经越过了叶子节点
2、方法二:辅助栈
因为要把每一个节点的左右节点调换位置,自然而然能想到先进后出的栈。
利用辅助栈来遍历二叉树所有的节点。
先将根节点入栈,再出栈,每次将栈弹出的元素的左右子节点,再交换左右子节点;然后弹出左子节点,再将它的左右子节点入栈,并且在树上交换左右子节点……然后弹出右子节点,重复着这个过程,直到栈中没有元素,此时二叉树遍历完成。
这里我可能讲得不是很清楚,附上力扣原作者的题解:
三、Java代码实现
1、递归
/**
* 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 root;
}
TreeNode temp = root.right;
root.right = mirrorTree(root.left);
root.left = mirrorTree(temp);
return root;
}
}
2、辅助栈
class Solution {
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;
}
return root;
}
}
四、时空复杂度分析
1、递归
时间复杂度: O(N) 遍历整个二叉树
空间复杂度: O(N) 最差情况下(当二叉树退化为链表),递归时系统需要O(N)大小的栈空间
2、辅助栈
时间复杂度 O(N) 其中 NN 为二叉树的节点数量,建立二叉树镜像需要遍历树的所有节点,占用 O(N)O(N) 时间。
空间复杂度 O(N) 如下图所示,最差情况下,栈 stackstack 最多同时存储(N+1)/2个节点,占用 O(N)额外空间。
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
文章由极客之音整理,本文链接:https://www.bmabk.com/index.php/post/125068.html