问题
翻转一个二叉树,例子如下:
输入:
4
/ \
2 7
/ \ / \
1 3 6 9
输出:
4
/ \
7 2
/ \ / \
9 6 3 1
可以看出相当于原二叉树的镜像,也就是像逆置二叉树的每一层
解决方法
递归
递归的交换左右节点,从上层到下层的顺序交换。
首先,交换根节点的左右节点,其左右节点的左右子树也会跟着过去
如下所示:
4
/ \
7 2
/ \ / \
6 9 1 3
然后递归进入下一层,交换其左右节点,可以看到就是,翻转完成。
可以这样理解:一开始翻转第二层的,他们的子树跟着过去,相当于翻转下一层的一半,就像一个数组,我们对其进行二分翻转,第一次找到中间位置,把数组分为两个部分,然后翻转,之后把左右部分接着在里面分成两个部分,对应着左右子树,翻转,直至翻转到最后只有两个元素组成的部分,翻转结束
代码如下:
class Solution {
public TreeNode invertTree(TreeNode root) {
if (root==null)
return null;
//翻转
TreeNode temp=root.right;
root.right=root.left;
root.left=temp;
invertTree(root.left);
invertTree(root.right);
return root;
}
}
非递归-层序遍历
根据上面给的思路,也可以不用递归进行翻转,用层序遍历的方法,遍历的时候翻转,对加入队列的节点的左右节点进行翻转,然后按照层序依次翻转。
思路参照上一个,只是换了个遍历方法,上面的递归也是一种遍历二叉树的方法
代码如下:
class Solution {
public TreeNode invertTree(TreeNode root) {
if (root == null)
return null;
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()){
TreeNode temp = queue.poll();
//交换
TreeNode change = temp.left;
temp.left = temp.right;
temp.right = change;
if (temp.left != null)
queue.add(temp.left);
if (temp.right != null)
queue.add(temp.right);
}
return root;
}
}