226. 翻转二叉树
以下是实现代码,运用了递归函数,将问题拆分成一个个可以解决的小问题。
/**
* 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){
return root;//如果是空树,不需要反转
}
changeLR(root);
return root;
}
TreeNode* changeLR(TreeNode* root){
if(root->left==NULL && root->right==NULL){
return root;
}
if(root->left!=NULL){
changeLR(root->left);
}
if(root->right!=NULL){
changeLR(root->right);
}
//交换左右节点
TreeNode* temp = root->left;
root->left = root->right;
root->right = temp;
return root;
}
};
运行结果如下,
内存消耗有点大。