描述
翻转一棵二叉树。
备注:
这个问题是受到 Max Howell 的 原问题 启发的 :
谷歌:我们90%的工程师使用您编写的软件(Homebrew),但是您却无法在面试时在白板上写出翻转二叉树这道题,这太糟糕了。
思路
按照《从三道LeetCode例题总结递归套路,解决递归问题》,,将二叉树抽象为只有一个root结点和其左右孩子的树,按步骤分析:
- 递归的出口:当root结点为空时,递归结束
- 应该返回什么信息:这道题要返回的是被交换过后的子树。
- 本级递归需要做什么:需要交换root结点的左右子树。注意:这里不是只交换结点的val值,而是交换root的左右指针指向的结点。放在整个二叉树中看,即将root结点的左右指针的指向调换,root->left指向root->right,root->right指向root->left。这样不光交换了左右子结点,也交换了左右子结点的子树。
解答
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if(root == NULL) return NULL;//1.递归出口
//3.本层递归需要做什么:交换root结点的左右孩子
TreeNode* tmp = root->left;
root->left = root->right;
root->right = tmp;
invertTree(root->left);
invertTree(root->right);
return root;//3.返回什么值
}
};