226. 翻转二叉树(迭代，递归)2

翻转一棵二叉树。

示例：

输入：

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

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

法一：递归

既然树中的每个节点都只被访问一次，那么时间复杂度就是 O(n)，其中 n 是树中节点的个数。在反转之前，不论怎样我们至少都得访问每个节点至少一次，因此这个问题无法做地比 O(n) 更好了。

本方法使用了递归，在最坏情况下栈内需要存放 O(h) 个方法调用，其中h是树的高度。由于 h\in O(n)h∈O(n)，可得出空间复杂度为 O(n)。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode invertTree(TreeNode root) {
        if (root == null) return null;

        TreeNode right = invertTree(root.right);
        TreeNode left = invertTree(root.left);
        root.left = right;
        root.right = left;
        return root;
    }
}

法二：迭代

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
	public TreeNode invertTree(TreeNode root) {
		if(root==null) return null;
		
		//将二叉树中的节点逐层放入栈中，再迭代处理栈中的元素
		Stack<TreeNode> stack = new Stack<>();
		stack.push(root);
		while(!stack.isEmpty()) {
			//每次都从栈中拿一个节点，并交换这个节点的左右子树
			TreeNode tmp = stack.pop();
			TreeNode left = tmp.left;
			tmp.left = tmp.right;
			tmp.right = left;
			//如果当前节点的左子树不为空，则放入栈等待后续处理
			if(tmp.left!=null) {
				stack.push(tmp.left);
			}
			//如果当前节点的右子树不为空，则放入栈等待后续处理
			if(tmp.right!=null) {
				stack.push(tmp.right);
			}
		}
		//返回处理完的根节点
		return root;
	}
}