题目描述
给你一棵二叉树的根节点 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;
}