LeetCode—226. 翻转二叉树

方式一：递归

我们下文以前序遍历为例，通过动画来看一下翻转的过程:

我们来看一下递归三部曲：

1、确定递归函数的参数和返回值

• 参数就是要传入节点的指针，不需要其他参数了，通常此时定下来主要参数，如果在写递归的逻辑中发现还需要其他参数的时候，随时补充。

• 返回值的话其实也不需要，但是题目中给出的要返回root节点的指针，可以直接使用题目定义好的函数，所以就函数的返回类型为TreeNode。

 public TreeNode invertTree(TreeNode root) {}

2、确定终止条件

• 当前节点为空的时候，就返回

if (root == NULL) return root;

3、确定单层递归的逻辑

• 因为是先前序遍历，所以先进行交换左右孩子节点，然后反转左子树，反转右子树。

swap(root->left, root->right);
invertTree(root->left);
invertTree(root->right);

也就是交换当前节点的左右节点，再递归的交换当前节点的左节点，递归的交换当前节点的右节点。

//DFS递归
class Solution {
   /**
     * 前后序遍历都可以
     * 中序不行，因为先左孩子交换孩子，再根交换孩子（做完后，右孩子已经变成了原来的左孩子），再右孩子交换孩子（此时其实是对原来的左孩子做交换）
     */
    public TreeNode invertTree(TreeNode root) {
    	//递归函数的终止条件，节点为空时返回
        if (root == null) {
            return null;
        }
        //将当前节点的左右子树交换
        swapChildren(root);
         //递归交换当前节点的 左子树
        invertTree(root.left);
        //递归交换当前节点的 右子树
        invertTree(root.right);
        //函数返回时就表示当前这个节点，以及它的左右子树都已经交换完了
        return root;
    }

    private void swapChildren(TreeNode root) {
    	//下面三句是将当前节点的左右子树交换
        TreeNode tmp = root.left;
        root.left = root.right;
        root.right = tmp;
    }
}

方式二：迭代
递归实现也就是深度优先遍历的方式，那么对应的就是广度优先遍历。 广度优先遍历需要额外的数据结构–队列，来存放临时遍历到的元素。 深度优先遍历的特点是一竿子插到底，不行了再退回来继续；而广度优先遍历的特点是层层扫荡。 所以，我们需要先将根节点放入到队列中，然后不断的迭代队列中的元素。对当前元素调换其左右子树的位置，然后：

• 判断其左子树是否为空，不为空就放入队列中
• 判断其右子树是否为空，不为空就放入队列中
  动态图如下：

深度优先遍历和广度优先遍历，从动画图中看起来很类似，这是因为演示的树层数只有三层。
时间复杂度：同样每个节点都需要入队列/出队列一次，所以是 O(n)空间复杂度：最坏的情况下会包含所有的叶子节点，完全二叉树叶子节点是 n/2个，所以时间复杂度是 0(n)代码实现如下：

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