Leetcode C++《热题 Hot 100-12》226.翻转二叉树

Leetcode C++《热题 Hot 100-12》226.翻转二叉树

1. 题目
   翻转一棵二叉树。

示例：

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

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

2. 思路

• 递归大法好理解，递归时间复杂度虽然是n，但是空间复杂度有可能很高，最极端情况递归栈包含了O(height)个函数
• 递归优化的方法，一般是将其转换成迭代，我们来分析这个题目是否可以转换成迭代
• 将没有转换左右孩子的依次加入到队列，然后逐个处理，直到队列为空
• 迭代时间复杂度、空间复杂度都是O(n)，棒！

3. 代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        //一道做过的题目，递归大法好
        /*if (root == NULL)
            return NULL;
        TreeNode* leftChild = root->left;
        TreeNode* rightChild = root->right;
        root->left = invertTree(rightChild);
        root->right = invertTree(leftChild);
        return root;*/
        //但递归时间复杂度虽然是n，但是空间复杂度有可能很高，最极端情况递归栈包含了O(height)个函数
        //递归优化的方法，一般是将其转换成迭代，我们来分析这个题目是否可以转换成迭代
        //将没有转换左右孩子的依次加入到队列，然后逐个处理，直到队列为空
        queue<TreeNode*> needReversedNode;
        if (root == NULL)
            return NULL;
        needReversedNode.push(root);
        while(!needReversedNode.empty()) {
            TreeNode* temp = needReversedNode.front();
            needReversedNode.pop();
            TreeNode* oldLeft = temp->left;
            temp->left = temp->right;
            temp->right = oldLeft;
            //temp->left!= NULL 不能用(!temp->left)替代，后者语法不对
            if (temp->left != NULL)
                needReversedNode.push(temp->left);
            if (temp->right != NULL)
                needReversedNode.push(temp->right);
        }
        return root;
    }
};