代码随想录算法训练营第二十三天 | 669. 修剪二叉搜索树、108. 将有序数组转换为二叉搜索树、538. 把二叉搜索树转换为累加树
669. 修剪二叉搜索树
题目
解法
- 递归法
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int low, int high) {
if (root == nullptr) return nullptr; // 下一个节点为空,则结束探索
if (root->val < low) { // 如果在第一轮递归中,会把根节点改变
TreeNode* right = trimBST(root->right, low, high); // 接入满足条件的右子树
return right;
}
if (root->val > high) { // 如果在第一轮递归中,会把根节点改变
TreeNode* left = trimBST(root->left, low, high); // 接入满足条件的左子树
return left;
}
root->left = trimBST(root->left, low, high); // 符合范围的节点
root->right = trimBST(root->right, low, high);// 符合范围的节点
return root;
}
};
递归逻辑理解
108. 将有序数组转换为二叉搜索树
题目
解法
class Solution {
public:
TreeNode* traversal(vector<int>& nums, int left, int right) {
if(left > right) return nullptr;
int mid = left + ((right - left) / 2);
TreeNode* root = new TreeNode(nums[mid]);
root->left = traversal(nums, left, mid-1);
root->right = traversal(nums, mid+1, right);
return root;
}
TreeNode* sortedArrayToBST(vector<int>& nums) {
return traversal(nums, 0, nums.size()-1);
}
};
538. 把二叉搜索树转换为累加树
题目
解法
- 递归 :右中左遍历
class Solution {
private:
int pre = 0; //保存上一个值
void traversal(TreeNode* cur) {
if (cur == nullptr) return;
traversal(cur->right); // 右中左顺序
cur->val += pre;
pre = cur->val;
traversal(cur->left);
}
public:
TreeNode* convertBST(TreeNode* root) {
traversal(root);
return root;
}
};
感悟
花时间把一些模糊的概念弄清楚很有必要,比如递归的逻辑,把过程一步一步写出来,这样对于递归的理解和把握就比较清晰了。