[leetcode]226.翻转二叉树

翻转一棵二叉树。

示例:

输入:

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

     4
   /   \
  7     2
 / \   / \
9   6 3   1

备注:
这个问题是受到 Max Howell 的 原问题 启发的 :

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

思路1:
dfs先序遍历二叉树,递归翻转左子树,右子树。

AC代码:

class Solution {
   public:
    TreeNode* invertTree(TreeNode* root) {
        dfs(root);
        return root;
    }
    void dfs(TreeNode* root) {
        if (root) {
            swap(root->left, root->right);
            dfs(root->left);
            dfs(root->right);
        }
    }
};

思路2:
bfs层序遍历二叉树,每次出队一个节点,反转其左右子树之后将孩子节点入队。
AC代码:(C++)

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

你可能感兴趣的:(LeetCode)