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); // 寻找符合区间[low, high]的节点 return right; } if (root->val > high) { TreeNode* left = trimBST(root->left, low, high); // 寻找符合区间[low, high]的节点 return left; } root->left = trimBST(root->left, low, high); // root->left接入符合条件的左孩子 root->right = trimBST(root->right, low, high); // root->right接入符合条件的右孩子 return root; } };
思路
-
Base Case:
-
if (root == nullptr) return nullptr;
:如果当前节点为空,则返回nullptr
。这是递归的基准情况,意味着已经到达了树的叶子节点下的空位置,或者是一开始就给定了一个空树。
-
-
修剪当前节点:
-
如果当前节点的值小于
low
,说明当前节点及其左子树的所有节点都不在范围内,因为在二叉搜索树中,一个节点的左子树都比该节点的值小。
-
TreeNode* right = trimBST(root->right, low, high);
:递归调用trimBST
来修剪当前节点的右子树,因为可能存在符合条件的节点。 -
return right;
:返回修剪后的右子树。这里,当前节点及其左子树被舍弃,因为它们的值不在[low, high]范围内。
-
-
如果当前节点的值大于
high
,类似地,说明当前节点及其右子树的所有节点都不在范围内。
-
TreeNode* left = trimBST(root->left, low, high);
:递归调用trimBST
来修剪当前节点的左子树。 -
return left;
:返回修剪后的左子树。这里,当前节点及其右子树被舍弃。
-
-
-
递归修剪左右子树:
-
如果当前节点的值在[low, high]范围内,那么需要对其左右子树进行修剪,保证左右子树中的所有节点也满足这个条件。
-
root->left = trimBST(root->left, low, high);
:递归修剪左子树,确保左子树中的所有节点的值也在[low, high]范围内。 -
root->right = trimBST(root->right, low, high);
:递归修剪右子树。
-
-
return root;
:返回修剪后的当前节点,它作为新的树(或子树)的根节点。
-
108.将有序数组转换为二叉搜索树*
代码
示例代码
class Solution { private: 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; } public: TreeNode* sortedArrayToBST(vector<int>& nums) { TreeNode* root = traversal(nums, 0, nums.size() - 1); return root; } };
思路
构造时,每次取数组中间值,前序遍历
538.把二叉搜索树转换为累加树*
代码
示例代码
class Solution { public: int pre = 0; // 记录前一个节点的数值 void traversal(TreeNode* cur) { // 右中左遍历 if (cur == NULL) return; traversal(cur->right); cur->val += pre; pre = cur->val; traversal(cur->left); } TreeNode* convertBST(TreeNode* root) { pre = 0; traversal(root); return root; } };
思路
反中序遍历