翻转二叉树
翻转一棵二叉树。
示例:
输入:
4
/ \
2 7
/ \ / \
1 3 6 9
输出:
4
/ \
7 2
/ \ / \
9 6 3 1
方法1:递归
思路:其实就是交换以下左右节点,然后再递归的交换左节点,右节点
终止条件:当前节点为null时返回
交换当前节点的左右节点,再递归交换当前节点的左节点,递归的交换当前节点的右节点
时间复杂度:每个元素都必须访问一次,使用O(n)
空间复杂度:最坏情况下,需要存放O(h)个函数调用
class Solution{
public TreeNode invertTress(TreeNode root){
//递归终止条件,节点为空时返回
if(root==null){
return null;
}
TreeNode tmp = root.right;
root.right=root.left;
root.left=tmp;
//递归交换当前节点的左子树
invertTree(root.left);
//递归交换当前节点的右子树
inverTree(root.right);
//函数返回时就表示当前这个节点,以及它的左右子树
return root;
}
}
方法2:迭代
迭代就是广度优先遍历,额外需要的数据结构是队列,来存放临时遍历的元素
先将根节点放入队列中,然后不断地迭代队列中地元素
对当前地元素调换其左右子树地位置
- 判断其左子树是否为空,不为空就放入队列
- 判断其右子树是否为空,不为空就放入队列
class Solution {
public TreeNode invertTree(TreeNode root) {
if(root==null) {
return null;
}
//将二叉树中的节点逐层放入队列中,再迭代处理队列中的元素
LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
queue.add(root);
while(!queue.isEmpty()) {
//每次都从队列中拿一个节点,并交换这个节点的左右子树
TreeNode tmp = queue.poll();
TreeNode left = tmp.left;
tmp.left = tmp.right;
tmp.right = left;
//如果当前节点的左子树不为空,则放入队列等待后续处理
if(tmp.left!=null) {
queue.add(tmp.left);
}
//如果当前节点的右子树不为空,则放入队列等待后续处理
if(tmp.right!=null) {
queue.add(tmp.right);
}
}
//返回处理完的根节点
return root;
}
}