本文主要是介绍代码随想录算法训练营DAY23|C++二叉树Part.9|669.修建二叉搜索树、108.将有序数组转换为二叉搜索树、538.把二叉搜索树转换为累加树，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
文章目录
- 669.修建二叉搜索树
- 递归法思路
- 伪代码
- CPP代码
- 108.将有序数组转换为二叉搜索树
- 递归伪代码
- CPP代码
- 538.把二叉搜索树转换为累加树
- 思路
- 递归伪代码
- 递归CPP代码
- 迭代法
669.修建二叉搜索树
力扣题目链接
文章讲解：669.修建二叉搜索树
视频讲解：你修剪的方式不对，我来给你纠正一下！| LeetCode：669. 修剪二叉搜索树
状态：这个最关键的点在于，删除结点之后应该如何接上去
递归法思路
本题中最关键的在于我们应该如何重构BST，其实跟我们之前讲的题目删除二叉树中的结点类似。
这题一定要画图模拟才能搞清楚。
伪代码
- 确定递归函数的参数和返回值：本题中有返回值可以大大降低代码难度，因为我们可以通过递归函数的返回值移除结点。
TreeNode* trimBST(TreeNode* root, int low, int high)
- 确定终止条件：修剪的操作并不是在终止条件上进行的，所以遇到空结点返回即可。
if (root == nullptr) return nullptr;
-
单层递归逻辑
- 如果root(当前结点)的元素小于low的数值，那么应该递归右子树，并返回右子树符合条件的头结点。
if (root->val < low){TreeNode* right = trimBST(root->right, low, high);return right; }- 如果root(当前结点)的元素大于high，那么应该递归左子树，并返回左子树符合条件的头结点
if (root->val > high){TreeNode* left = trimBST(root->left, low, high);return left }- 将下一层处理完左子树的结果赋值给root->left，处理完右子树的结果赋给root->right
root->left = trimBST(root->left, low, high); // root->left接入符合条件的左孩子 root->right = trimBST(root->right, low, high); // root->right接入符合条件的右孩子 return root;
CPP代码
#整体代码
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;}
};
#精简代码
class Solution {
public:TreeNode* trimBST(TreeNode* root, int low, int high) {if (root == nullptr) return nullptr;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.将有序数组转换为二叉搜索树
力扣题目链接
文章讲解：108.将有序数组转换为二叉搜索树
视频讲解：构造平衡二叉搜索树！| LeetCode：108.将有序数组转换为二叉搜索树
状态：easy模式，其实就是构造二叉树。但是其中的难点在于，如何保持平衡呢？
递归伪代码
- 确定递归函数返回值和参数:无论是删除还是增加二叉树的结点，都是用递归函数的返回值来完成的，这样操作会比较方便。
本题同理，依然用递归函数u的返回值来构造中结点的左右孩子。
在构造二叉树的时候尽量不要重新定义左右区间数组，而是用下标来操作原数组
//左闭右闭区间[left, right]
TreeNode* traversal(vector<int>& nums, int left, int right)
在这里，我们定义好了左闭右闭区间，为了遵守循环不变量原则，在不断分割的过程中也一定要坚持左闭右闭的原则
- 确定递归终止条件
这里定义的是左闭右闭的区间，所以当区间left>right当时候，就是空结点了
if (left > right) return nullptr;
- 确定单层递归逻辑
划分区间的时候，root的左孩子接住下一层左区间的构造结点，右孩子接住下一层右区间构造的结点。
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;
CPP代码
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.把二叉搜索树转换为累加树
力扣题目链接
文章讲解：538.把 二叉搜索树转换为累加树
视频讲解：普大喜奔！二叉树章节已全部更完啦！| LeetCode：538.把二叉搜索树转换为累加树
状态： 每个结点都把比他大的结点做一个相加成为新结点值。感觉真的是毫无头绪
思路
直接在二叉树上操作真的很难。但是如果转换思路成一个数组呢？
给定一个有序数组，我们要将这个数组变成累加数组。
也就是说数组上的每个位置，都把其前面位置做一个相加，瞬间就能变成一个送分题
就是一个有序数组[2, 5, 13]，求从后到前的累加数组，也就是[20, 18, 13]，是不是感觉这就简单了，这里我们对数组的操作要从后向前进行遍历。
那么如果反应到二叉搜索树呢？
我们用中序遍历的二叉搜索树是有序数组，那么我们进行一个反中序遍历，也就是遍历顺序为右中左，然后顺序累加即可。那么如何实现这一块代码呢？换个顺序即可
递归伪代码
- 递归函数参数和返回值:在这里我们不需要任何会返回值了，最主要的任务就是遍历整颗树
int pre = 0;
void traversal (TreeNode* cur)
- 确定终止条件
if (cur == NULL) return;
- 单层递归逻辑
右中左遍历二叉树，中结点的处理逻辑就是让cur的数值加上前一个结点的数值。
traversal(cur->right);
cur->cur += pre;
pre = cur->val;
traversal(cur->left;)
递归CPP代码
class Solution {
private:int pre = 0; // 记录前一个节点的数值void traversal(TreeNode* cur) { // 右中左遍历if (cur == NULL) return;traversal(cur->right);cur->val += pre;pre = cur->val;traversal(cur->left);}
public:TreeNode* convertBST(TreeNode* root) {pre = 0;traversal(root);return root;}
};
迭代法
class Solution {
private:int pre; // 记录前一个节点的数值void traversal(TreeNode* root) {stack<TreeNode*> st;TreeNode* cur = root;while (cur != NULL || !st.empty()) {if (cur != NULL) {st.push(cur);cur = cur->right; // 右} else {cur = st.top(); // 中st.pop();cur->val += pre;pre = cur->val;cur = cur->left; // 左}}}
public:TreeNode* convertBST(TreeNode* root) {pre = 0;traversal(root);return root;}
};
这篇关于代码随想录算法训练营DAY23|C++二叉树Part.9|669.修建二叉搜索树、108.将有序数组转换为二叉搜索树、538.把二叉搜索树转换为累加树的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
