leetcode226.翻转二叉树

题目

给你一棵二叉树的根节点 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

解答

• 深度优先遍历

使用dfs函数辅助遍历过程。代码如下：

/**
 * 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 == NULL || (root->left == NULL && root->right == NULL)) { 
            // 如果root为空或者root左右子树都为空，则直接返回root
            return root;
        }
        dfs(root); // 否则，则对root进行深度优先遍历
        return root;
    }
    void dfs(TreeNode* node) {
        if (node->left == NULL && node->right == NULL) {
            return; // 如果被遍历的结点左右子树都为空，则直接跳出
        }
        TreeNode* rightCopy = node->right; // 新建一个节点rightCopy用于复制当前被遍历结点的右子树
        node->right = node->left; // 将节点的左子树直接赋值给右子树
        node->left = rightCopy; // 将rightCopy赋值给左子树，完成当前结点的交换
        if (node->left != NULL) { // 如果当前结点的左子树不为空，则对该子树进行遍历
            dfs(node->left);
        }
        if (node->right != NULL) { // 如果当前结点的右子树不为空，则对该子树进行遍历
            dfs(node->right);
        }
    }
};