Hot100-Day43-T226翻转二叉树

Day43[26/4/12]T226翻转二叉树

给你一棵二叉树的根节点 root ，翻转这棵二叉树，并返回其根节点。

示例 1：

输入：root = [4,2,7,1,3,6,9]
输出：[4,7,2,9,6,3,1]

示例 2：

输入：root = [2,1,3]
输出：[2,3,1]

示例 3：

输入：root = []
输出：[]

提示：

• 树中节点数目范围在 [0, 100] 内
• -100 <= Node.val <= 100

解题思路

递归翻转就完事了。

先对子树进行翻转，左右子树都需要。

然后对自身进行翻转，就是交换左右节点。

Code

#include <algorithm>

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)
    {
        // 1. 先处理 nullptr
        if (root == nullptr)
        {
            return nullptr;
        }

        // 先递归交换
        invertTree(root->left);
        invertTree(root->right);

        // 再对自己交换
        std::swap(root->left, root->right);

        return root;
    }
};