【Leetcode】226. 翻转二叉树

QUESTION

easy

题目描述

翻转一棵二叉树。

示例：

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

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

备注:

这个问题是受到 Max Howell 的 原问题 启发的：

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

说明

无

---

SOLUTION

看到备注，相信这个老哥崩溃了

方法一（递归）

这道题其实很好写，我被谷歌吓到了。

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if (!root) return NULL;
        TreeNode *tmp = root->left;
        root->left = invertTree(root->right);
        root->right = invertTree(tmp);
        return root;
    }
};

方法二（非递归）

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        queue<TreeNode*> q;
        if(root) q.push(root);
        while(!q.empty()){
            TreeNode *curr = q.front();
            if(curr->left) q.push(curr->left);
            if(curr->right) q.push(curr->right);
            TreeNode *tmp = curr->left;
            curr->left = curr->right;
            curr->right = tmp;
            q.pop();
        }
        return root;
    }
};