LeetCode Algorithms 226. Invert Binary Tree 题解

题目：反转二叉树
Trivia:
This problem was inspired by this original tweet by Max Howell:
Google: 90% of our engineers use the software you wrote (Homebrew), but you can’t invert a binary tree on a whiteboard so fuck off.

原来大名鼎鼎的max没手写出来的反转二叉树是这样的！让我们来看看这道题怎么做吧

首先把一棵树反转，先想到要按中序遍历来反转，因为这样反转可以保证反转之后顺序是反转的

我们也可以这样想，反转一棵二叉树，那么每个节点都要反转一次，如果从根节点开始反转（先序遍历），会不会有不一样的结果呢。答案是不会，结果还是符合题目要求的反转二叉树。原因很简单，每个节点都反转一次，无论是先序遍历还是中序遍历还是后序遍历，都只访问一次节点，所以节点都是只反转一次的，所以结果是一样的。

因为先序遍历实现比较简单，所以我的实现代码也是按先序遍历来写的：

TreeNode* invertTree(TreeNode* root) {
    if (root == NULL) return root;
    stack<TreeNode* > s;
    TreeNode* current = root;
    s.push(current);
    while (!s.empty()) {
        current = s.top();
        s.pop();
        swap(current->left, current->right);
        if (current->left != NULL) {
            s.push(current->left);
        }
        if (current->right != NULL) {
            s.push(current->right);
        }
    }
    return root;
}