给你一棵二叉树的根节点 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 = [] 输出:[]
/**
* 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) { // invertTree 函数,用于翻转二叉树
// 如果当前节点为空,则直接返回空指针
if(root==nullptr)
return nullptr;
// 递归地翻转左子树,并将翻转后的子树存储在 left 变量中
auto left = invertTree(root->left);
// 递归地翻转右子树,并将翻转后的子树存储在 right 变量中
auto right = invertTree(root->right);
// 交换当前节点的左右子节点
root->right = left;
root->left = right;
// 返回翻转后的根节点
return root;
}
};
首先,代码中给出了二叉树节点的定义,每个节点包含一个值和指向左右子节点的指针。
接下来,给出了一个名为invertTree
的函数,用于翻转二叉树。该函数接受一个指向根节点的指针,并返回翻转后的二叉树的根节点。
在函数内部,首先进行了边界条件的判断,如果当前节点为空,直接返回空指针。
然后,使用递归的方式翻转左子树和右子树。分别调用invertTree
函数递归地翻转左子树和右子树,并将翻转后的子树分别存储在left
和right
变量中。
最后,交换当前节点的左右子节点,将翻转后的左子树赋给当前节点的右子节点,将翻转后的右子树赋给当前节点的左子节点。
最后,返回翻转后的根节点。
整个过程通过递归的方式实现了二叉树的翻转。