上次说了很多用递归可以简单解决的二叉树题目,这次继续用一些题目说明,同时探究一下递归中的回溯现象。
继续用递归解题
由浅及深,先来这道题。
给定两个二叉树将它们合并,合并就是互补的过程,如果两棵树的对应位置都有节点,那么就将值相加。
我们可以先合并根节点,再到达子树,也可以反过来。
所以遍历顺序是无关的。
我们拿后序来举例,需要思考的是如果我们已经将根节点的左右子树合并成功,根节点如何操作得到的结果。
当然操作其实也很简单,就是把两个根节点的值相加,然后把合并后的左右子树连到根节点上。
TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
if(root1==NULL)
return root2;
if(root2==NULL)
return root1;
TreeNode *left=mergeTrees(root1->left,root2->left);
TreeNode *right=mergeTrees(root1->right,root2->right);
root1->val+=root2->val;
root1->left=left;
root1->right=right;
return root1;
}
再来一道
给定一个二叉树,返回所有左叶子之和。
左叶子,需要翻译一下,就是左边没有左右子树的叶子节点。
我们使用后序遍历从叶子节点开始,计算左叶子,然后递归将它们相加,就是左叶子之和。
用我们一贯的思路来看,就是在得到左右子树的左叶子之和后,从根节点出发,如何加上根节点的结果。
需要注意的是左叶子的判断,只有符合条件的才可以加入结果。
int sumOfLeftLeaves(TreeNode* root) {
if(root==NULL)
return 0;
int left=sumOfLeftLeaves(root->left);
int right=sumOfLeftLeaves(root->right);
int value=0;
if(root->left&&!root->left->left&&!root->left->right){
value=root->left->val;
}
return value+left+right;
}
还有一道经典的题目,平衡二叉树
平衡二叉树的概念我们都很熟悉,一个二叉树每个节点的左右两个子树的高度差的绝对值不超过1
这道题要求判断一个数是否为平衡二叉树。
出发点就是先判断子树,再递归到根节点。如果子树左右高度差已经大于1,那么可以直接返回false
但是我们需要高度差是int值,而且需要这个int值来进行递归,所以需要辅助函数。
int getHeight(TreeNode *root){
if(root==NULL)
return 0;
int left=getHeight(root->left);
if(left==-1)return -1;
int right=getHeight(root->right);
if(right==-1)return -1;
int result;
if(abs(left-right)>1){
return -1;
}
else{
result=max(left,right)+1;//返回最高的子树高度
}
return result;
}
bool isBalanced(TreeNode* root) {
return getHeight(root)==-1?false:true;
}
再来一道层序遍历中比较常见的题目,我们用递归来解答