题目:翻转二叉树. - 力扣(LeetCode)
题解:递归
题目很简单,需要翻转所有得节点得左右子树。对于树类问题,我们当然首先考虑递归。
在递归过程,我们要考虑得终止条件,而这里得终止条件就是叶子节点。当节点为叶子节点时,其本身并不会进行翻转。而如何进行翻转,实际上与swap函数同理,那么剩下只是一个递归过程。
考虑到一棵树,有递归有四种情况:
- 无左孩子有右孩子:root->left 为 nullptr,root->rigtht 不为nullptr(只递归右侧)
- 无右孩子有左孩子:root->rigtht 为 nullptr,root->left 不为nullptr(只递归左侧)
- 左右孩子都有:root->left 、root->rigtht 均不为nullptr(递归左右侧)
- 左右孩子都无:root->left 、root->rigtht 均为nullptr(叶子节点,返回)
代码:
/**
* 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* reverse(TreeNode* root)
{
if (root == nullptr) // 检查空指针
return nullptr;
if(root->left == nullptr && root->right == nullptr) //判断是否是叶子节点,直接返回
return root;
if(root->left != nullptr) reverse(root->left);
if(root->right != nullptr) reverse(root->right);
TreeNode* p = root->left;
root->left = root->right;
root->right = p;
return root;
}
TreeNode* invertTree(TreeNode* root) {
TreeNode* reverse_Tree = reverse(root);
return reverse_Tree;
}
};
思考:对于树得递归
对于树得递归来说,我们只需要考虑到叶子节点,或者上几层来进行模拟,来处理题目中进行过程得操作。剩下的只是判断左右子树是否递归,