修剪二叉搜索树
修剪一棵二叉搜索树,将节点node不在low和hig范围内的删除。们确定递归函数的参数和返回值,传入参数为节点node,最小边界low,最大边界high。
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int low, int high) {
if(root == NULL) return NULL;
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;
}
};
这道题我们采取任意一种递归顺序都可以,我写的是中左右,我们先处理一个节点,如果一个节点不在[low,high]范围内,我们返回是它修剪结束之后子树的根节点。然后我们处理左子树,对于例题的0不在[1,3]范围内,我们return的就是0的子树2->1的根节点2,我们用root->left = trimBST(root->left,low,high)来处理root的左子树,那么root->left接住的就是修剪掉0之后的子树,这样就完成了修剪操作。
将有序数组转换成二叉搜索树
将一个有序数组转换成二叉搜索树,我们选取数组的中间数作为二叉搜索树的中节点,我们取二叉搜索树的起始节点指标为left,最后一个节点坐标为right。返回值为二叉树的中节点,中节点的左节点为子树的根坐标,中节点的右节点为子树的根左边。
class Solution {
private:
TreeNode* traversal(vector<int>& nums,int left,int right)
{
if(left > right) return NULL;
int mid = left + (right - left) / 2;
TreeNode* node = new TreeNode(nums[mid]);//以数组中间节点为根节点
node->left = traversal(nums,left,mid-1);
node->right = traversal(nums,mid+1,right);
return node;
}
public:
TreeNode* sortedArrayToBST(vector<int>& nums) {
return traversal(nums,0,nums.size()-1);
}
};
把二叉搜索树转换成累加树
这道题可以类比成一个数组从后往前遍历相加,我们选取两个指针用来标记当前节点的值和前一个节点的值。并且这道题是从后往前,我们采取中序的倒序来进行解题,也就是右中左。
class Solution {
private:
int pre = 0;
void traversal(TreeNode* node){
if(node == NULL) return;
traversal(node->right);//右
node->val += pre;
pre = node->val;
traversal(node->left);//左
}
public:
TreeNode* convertBST(TreeNode* root) {
traversal(root);
return root;
}
};