Leetcode刷题226. 翻转二叉树

翻转一棵二叉树。

示例：

输入：

     4
   /   \
  2     7
 / \   / \
1   3 6   9
输出：

     4
   /   \
  7     2
 / \   / \
9   6 3   1
备注:
这个问题是受到 Max Howell 的 原问题 启发的 ：

谷歌：我们90％的工程师使用您编写的软件(Homebrew)，但是您却无法在面试时在白板上写出翻转二叉树这道题，这太糟糕了。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/invert-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

感谢 labuladong的详细解法，传送门东哥手把手带你刷二叉树（思维篇）

class Solution {
    public TreeNode invertTree(TreeNode root) {
//		return invertTreeI(root);
//		return invertTreeII(root);
//		return invertTreeIII(root);
		return invertTreeIIII(root);
	}

	//方法四：使用BFS遍历，时间和空间复杂度O(N)
	private TreeNode invertTreeIIII(TreeNode root) {
		if (root == null) {
			return null;
		}
		Queue<TreeNode> queue = new LinkedList<>();
		queue.offer(root);

		while (!queue.isEmpty()) {
			TreeNode node = queue.poll();

			TreeNode temp = node.left;
			node.left = node.right;
			node.right = temp;

			if (node.left != null) {
				queue.offer(node.left);
			}
			if (node.right != null) {
				queue.offer(node.right);
			}
		}
		return root;
	}

	//方法三：使用DFS遍历二叉树，在遍历过程中翻转二叉树
	//时间复杂度O(N)，空间复杂度O(N)
	private TreeNode invertTreeIII(TreeNode root) {
		if (root == null) {
			return null;
		}
		Stack<TreeNode> stack = new Stack<>();
		stack.push(root);

		while (!stack.isEmpty()) {
			TreeNode node = stack.pop();

			TreeNode temp = node.left;
			node.left = node.right;
			node.right = temp;

			if (node.right != null) {
				stack.push(node.right);
			}
			if (node.left != null) {
				stack.push(node.left);
			}
		}
		return root;
	}

	//方法二：定义递归函数invertTree，表示翻转以root为根节点的二叉树，并返回翻转后的根节点
	//时间和空间复杂度O(N)
	private TreeNode invertTreeII(TreeNode root) {
		//终止条件
		if (root == null) {
			return null;
		}
		//单独的某一节点，需要将节点的左右子树翻转
		TreeNode left = invertTreeII(root.left);
		TreeNode right = invertTreeII(root.right);

		//然后交换左右子节点
		root.left = right;
		root.right = left;

		//翻转后，返回根节点
		return root;
	}

	//方法一：定义traverse函数，遍历每个节点，让每个节点的左右子节点翻转
	//时间和空间复杂度O(N)
	private TreeNode invertTreeI(TreeNode root) {
		traverse(root);
		return root;
	}

	private void traverse(TreeNode root) {
		if (root == null) {
			return;
		}
		//前序遍历
		//单独的一个节点左右子节点翻转
		TreeNode temp = root.left;
		root.left = root.right;
		root.right = temp;

		//遍历左右子树
		traverse(root.left);
		traverse(root.right);
	}
}