226. 翻转二叉树

226. 翻转二叉树

方法一：递归（自底向上）

class Solution {
    public TreeNode invertTree(TreeNode root) {
        if (root == null) {
            return null;
        }
        TreeNode left = invertTree(root.left);
        TreeNode right = invertTree(root.right);
        root.left = right;
        root.right = left;
        return root;
    }
}

方法二：递归（自顶向下）

class Solution {
    public TreeNode invertTree(TreeNode root) {
        //递归函数的终止条件， 节点为空时返回
        if (root == null) {
            return null;
        }
        //将当前节点的左右子树交换
        TreeNode tmp = root.right;
        root.right = root.left;
        root.left = tmp;
        //递归交换当前节点的左子树
        invertTree(root.left);
        //递归交换当前节点的右子树
        invertTree(root.right);
        //函数返回时就表示当前这个节点，以及它的左右子树
        //都已经交换完了
        return root;
    }
}

方法三：迭代

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