669. 修剪二叉搜索树
题目链接/文章讲解: 代码随想录
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;
}
修建二叉搜索树,需要利用二叉搜索树的特性,左子树小于父节点,右子树大于父节点。因此当前节点小于目标区域时,应该返回右子树(有可能满足条件);大于时应该返回左子树。
最后递归的求解左右子树即可完成。
108.将有序数组转换为二叉搜索树
TreeNode* sortedArrayToBST(vector<int>& nums) {
int size = nums.size();
if (size == 0) return 0;
TreeNode* node = new TreeNode(nums[size/2]);
vector<int> left(nums.begin(), nums.begin()+size/2);
node->left = sortedArrayToBST(left);
vector<int> right(nums.begin()+size/2+1, nums.end());
node->right = sortedArrayToBST(right);
return node;
}
自己的递归做法,由于是平衡二叉搜索树,所以数组中间的元素肯定是二叉树的根节点(这样才能保持平衡),因此,利用数组中间元素来构建二叉树。因为要从上向下构建,要采用中序遍历。首先通过中间元素来当做根节点,然后使用左侧数组来构建左子树,使用右侧数组来构建右子树,这样就能构建出二叉搜索数了。
538.把二叉搜索树转换为累加树
class Solution {
public:
int sum = 0;
TreeNode* convertBST(TreeNode* root) {
if (root == NULL) return NULL;
convertBST(root->right);
sum += root->val;
root->val = sum;
convertBST(root->left);
return root;
}
};
我使用了一个全局变量来记录累加的和,然后进行右中左的中序遍历,从最右下的节点一路累加,直至加完整棵树即可得到累加树。