力扣 226. 翻转二叉树

这道题要求翻转二叉树所有节点，并返回其根节点。其实就是把每一个节点的左右子节点交换一下就可以了。题解采用前序遍历迭代法和层序遍历，但还有多种方法可以使用。

前序遍历迭代法时间复杂度：O(n)，空间复杂度：O(n)

前序遍历迭代法思路：

1. 确定递归函数的参数和返回值：参数就是要传入节点的指针，题目要求返回root节点的指针，所以直接利用题目定义好的函数作为递归函数。
2. 确定终止条件：当前节点为空的时候，就返回。
3. 确定单层递归的逻辑：因为是前序遍历，所以先进行左右子节点交换，然后递归反转左子树和反转右子树。

前序遍历迭代法代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if (root == nullptr)
            return root;
        swap(root->left, root->right);
        invertTree(root->left);
        invertTree(root->right);
        return root;
    }
};

层序遍历时间复杂度：O(n)，空间复杂度：O(n)

层序遍历思路：

• 层数遍历也可以翻转这棵树，因为层序遍历可以在遍历每个节点的时候将左右孩子都翻转一遍。

层序遍历代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        while (!que.empty()) {
            int size = que.size();
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                swap(node->left, node->right);
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
        }
        return root;
    }
};