Day23.修剪二叉搜索树、把二叉树转换为累加树
0669.修剪二叉搜索树
链接:0669.修剪二叉搜索树
- 如果当前节点为空,直接返回空
- 如果当前值小于low,那么当前节点和当前节点的左子树都要剪掉。注意不能直接返回右子树,因为右子树的左子树里还可能存在小于low的数。应当返回修剪后的右子树。
- 如果当前值大于hight,那么当前节点和当前节点的右子树都要剪掉。同样,由于左子树的右子树里还可能存在大于high的数,应当返回修剪后的左子树。
- 如果当前节点在范围内,那么更新当前左子树为修剪后的左子树;更新右子树为修剪后的右子树。
注意:在LeetCode中,这题如果自己手动delete节点,释放内存,会报错。
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int low, int high)
{
if (root == nullptr) {
return nullptr;
}
if (root->val < low) {
// 修剪当前节点的右子树
// 不能直接返回右子树,因为右子树的左子树里还可能存在小于low的数
TreeNode* right = trimBST(root->right, low, high);
// 删除当前节点的左子树
destroyTree(root->left);
// 删除当前节点
delete root;
return right;
} else if (root->val > high) {
// 修建当前节点的左子树
// 不能直接返回左子树,因为左子树的右子树里还可能存在大于high的数
TreeNode* left = trimBST(root->left, low, high);
destroyTree(root->right);
delete root;
return left;
} else {
// 如果当前节点在区间内
// 当前左子树等于修剪后的左子树
root->left = trimBST(root->left, low, high);
// 当前右子树等于修剪后的右子树
root->right = trimBST(root->right, low, high);
return root;
}
}
void destroyTree(TreeNode* root)
{
if (root == nullptr) {
return;
}
destroyTree(root->left);
destroyTree(root->right);
delete root;
}
};
0108.将有序数组转换为二叉搜索树
为保证二叉搜索数高度平衡,每次需要取数组的中间节点作为根节点。然后前半部分数组为左子树。后半部分数组为右子树。
class Solution {
public:
TreeNode* sortedArrayToBST(vector<int>& nums)
{
if (nums.empty()) {
return nullptr;
}
size_t mid = nums.size() / 2;
TreeNode* cur = new TreeNode(nums[mid]);
vector<int> left(nums.begin(), nums.begin() + mid);
cur->left = sortedArrayToBST(left);
vector<int> right(nums.begin() + (mid + 1), nums.end());
cur->right = sortedArrayToBST(right);
return cur;
}
};
0538.把二叉搜索树转换为累加树
二叉搜索树的中序遍历为一个递增有序数组,如果依然保持中序,但是调换左右子树位置,那么结果就是递减的有序数组。
由于目标值需要累加,所以记录一个pre
节点,以便累加。
class Solution {
public:
TreeNode* convertBST(TreeNode* root)
{
if (root == nullptr) {
return nullptr;
}
root->right = convertBST(root->right);
if (_pre != nullptr) {
root->val += _pre->val;
}
_pre = root;
root->left = convertBST(root->left);
return root;
}
private:
TreeNode* _pre = nullptr;
};