LeetCode_226_翻转二叉树-优快云博客

本文链接：https://blog.youkuaiyun.com/qq_41108978/article/details/126053051

本文探讨了两种方法实现二叉树节点的反转，包括使用迭代的层序遍历与递归策略。通过实例展示了如何利用`swap`函数高效地交换节点及其子节点，以及如何在`Solution`类中应用这两种方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

迭代法（通过层序遍历）反转：

swap(x, y)用来交换x和y的值，不用判断cur->left和cur->right是否存在，因为swap可以交换null和数值。
传入cur之后交换cur的结点就是交换原root的结点。

/**
 * 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* cur = que.front();
                que.pop();
                swap(cur->left, cur->right);
                if (cur->left) que.push(cur->left); //这两行可交换
                if (cur->right) que.push(cur->right);
            }
        }
        return root;
    }
};

递归法反转：

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        queue<TreeNode*> que;
        if (root == NULL) return root;
        swap(root->left, root->right);
        invertTree(root->left); //这两行可交换
        invertTree(root->right);
        return root;
    }
};