Leetcode——226. 翻转二叉树

问题

翻转一个二叉树,例子如下:

输入:

     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;
    }
}

你可能感兴趣的:(leetcode,java,数据结构)