669. 修剪二叉搜索树
思路
- 终止条件
当遇到空节点时即可返回;
当当前节点数值不在所给区间内,则有两种情况:当前节点值大于区间最大值;当前结点值小于区间最小值。
这里有一个易错点,当不满足区间时,如当前节点值大于区间最大值,它右子树内不一定全都是不满足条件的结点,所以要继续向下递归寻找是否有满足题意的结点,并且保留返回。如下图,当0不满足时,不能直接变成null,因为0的右子树上的2和1满足条件 - 返回的值被root结点接收,具体代码模拟过程可以手动模拟一下
代码实现
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int low, int high) {
if(root==nullptr)return root;
if(root->val<low)return trimBST(root->right,low,high);
if(root->val>high)return trimBST(root->left,low,high);
root->left=trimBST(root->left,low,high);
root->right=trimBST(root->right,low,high);
return root;
}
};
108.将有序数组转换为二叉搜索树
思路
构造二叉树用区间最为方便(可惜我基础没打好第一反应没想起来呜呜呜),和中序后序构造二叉树类似。虽然构造出的结构可以有多种,但构造为平衡二叉树才是最好的结构,不然就没有意义了。
已知传入数组为有序数组,则取中间节点构造为树的的根节点,将区间内的数组数据均分到左右子树,当代表左右区间的left>right,递归终止。
代码实现
class Solution {
public:
int i=1;
TreeNode* traversal(vector<int> &nums,int left,int right){
if(left>right)return nullptr;
int mid=(left+right)/2;
TreeNode* node=new TreeNode(nums[mid]);
node->left=traversal(nums,left,mid-1);
node->right=traversal(nums,mid+1,right);
return node;
}
TreeNode* sortedArrayToBST(vector<int>& nums) {
return traversal(nums,0,nums.size()-1);
}
};
538.把二叉搜索树转换为累加树
思路
- 终止条件
当遇到叶子结点即roo为空时返回 - 遍历过程
因为累加过程为从下往上、从左到右依次累加,所以遍历顺序应该为右中左。中间节点处理过程应该是先累加后再记录当前结点,以便于在使用了保存结点后才更改其保存数据。这道题和“求二叉搜索树众数”、“二叉搜索树最小绝对差”的代码思路几乎一致。
代码实现
class Solution {
public:
TreeNode *pre=nullptr;
void traversal(TreeNode* root){
if(root==nullptr)return ;//终止条件
traversal(root->right);//向右遍历
if(pre)root->val+=pre->val;//中节点处理
pre=root;
traversal(root->left);//向左遍历
return ;
}
TreeNode* convertBST(TreeNode* root) {
traversal(root);
return root;
}
};
总结
二叉树的学习终于告一段落,太开心啦