力扣 226题 翻转二叉树 记录

题目描述

给你一棵二叉树的根节点 root ，翻转这棵二叉树，并返回其根节点。
        4
       / \
      2   7
     /\   /\
    1  3 6  9

示例 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 = []
输出：[]

思路

1、invertTree(TreeNode* root): 用于翻转二叉树的成员函数。

• if (root == NULL) return root;: 如果传入的节点为空，则直接返回。这是递归的基础情况，确保递归有终止条件。
• swap(root->left, root->right);: 交换当前节点的左右子节点。
• 递归调用: 对左右子树递归地调用
• invertTree，以翻转整棵树的所有子树。 return root;: 返回翻转后的树的根节点。

2、printInOrder：中序遍历输出节点值

• if (node == nullptr) return;: 如果当前节点为空，终止递归。
• 递归遍历左子树: 调用 printInOrder来遍历左子树。
• 打印当前节点的值: 输出当前节点值。
• 递归遍历右子树: 调用 printInOrder 来遍历右子树

完整代码

#include<iostream>

using namespace std;

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) return root;
        swap(root->left, root->right);
        invertTree(root->left);
        invertTree(root->right);
        return root;
    }
};

// 中序遍历打印出节点
void printInOrder(TreeNode* node)
{
    if(node == NULL) return;
    printInOrder(node->left);
    cout << node->val << " ";
    printInOrder(node->right);
}

int main()
{
    TreeNode* root = new TreeNode(4);
    root->left = new TreeNode(2);
    root->right = new TreeNode(7);
    root->left->left = new TreeNode(1);
    root->left->right = new TreeNode(3);
    root->right->left = new TreeNode(6);
    root->right->right = new TreeNode(9);

    Solution s;
    TreeNode* invertedRoot = s.invertTree(root);

    printInOrder(invertedRoot);
    cout << endl;
}