669. 修剪二叉搜索树
leetcode链接
代码随想录链接
一刷状态:未通过
思路
递归思路比较巧妙,先对不符合区间的节点继续遍历修剪,使用左右孩子去接裁剪后返回的子树。
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int low, int high) {
if(root==nullptr) return nullptr;
// 如果节点的值小于区间,则返回以root->right为根节点的裁剪后的子树
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.将有序数组转换为二叉搜索树
leetcode链接
代码随想录链接
一刷状态:通过
思路
构造二叉树的思路,分左右区间分别构造,注意区间的定义,本题使用左闭右闭。
递归的终止条件为非法区间。
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.把二叉搜索树转换为累加树
leetcode链接
代码随想录链接
一刷状态:通过
思路
右中左遍历,使用双指针法,从后往前累加,使用pre记录上一个节点的值,加到当前节点。
class Solution {
public:
int pre = 0;
void traversal(TreeNode* cur)
{
if(cur==nullptr) return;
traversal(cur->right);
cur->val += pre;
pre = cur->val;
traversal(cur->left);
}
TreeNode* convertBST(TreeNode* root) {
traversal(root);
return root;
}
};