669. 修剪二叉搜索树
这题最开始的想法是复用删除节点的那题的思路做,需要修改的部分就是要让程序删除完一个点后继续遍历,因为后续可能还有不符合条件的节点。但这样想也做复杂了。
这类题其实不用想用什么序遍历,用哪种方式只是为了更好的达成目的,不要被必须用前中后常用的固定的顺序束缚住了。
这题翻译过来,要做的是找到小于low的节点,删除该节点及其左子树,将右子树连到该节点的父节点;找到大于high的节点,删除该节点以及它的右子树,将其左子树连接到该节点的父节点。
可以将情况分解成以下3类:
在low,high区间内的继续找左右子树;
碰到小于low的,继续在其右子树上找是否还有不在区间的,最后回溯返回这颗右子树;
碰到大于high的,继续在其左子树上找是否还有不在区间的,最后回溯返回这颗左子树。
这用什么顺序都行,只要能遍历二叉树。
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;}
};
这题在力扣平台上跑有点小问题,如果删除节点的话会报错,但在本地的ide上运行是不会有问题的,这应该是平台的问题。以下是找到的解释的博客。力扣平台delete的问题:delete出现内存报错ERROR: AddressSanitizer: heap-use-after-free on address-CSDN博客
108.将有序数组转换为二叉搜索树
这题的收获是再次提醒我了做题时要统一区间的定义。
class Solution {
public:TreeNode* build(vector<int>& nums, int left, int right){if(left > right) return nullptr;int mid = (left + right) / 2;TreeNode* node = new TreeNode(nums[mid]);node->left = build(nums, left, mid-1);node->right = build(nums, mid+1, right);return node;}TreeNode* sortedArrayToBST(vector<int>& nums) {return build(nums, 0, nums.size()-1);}
};
538.把二叉搜索树转换为累加树
这题的收获是,不要被学过的知识束缚。一开始想到了,这题转成数组,从后往前累加即可,但对应到树上,发现使用学过的前中后怎么都不适合来遍历。看完题解,才发现还能 右中左 这样遍历,被学过的知识锁住了,其实这很简单就能想到,左中右 中序遍历,能从小到大输出,那么 右中左 则是能从大到小输出。前提是这是二叉搜索树。
class Solution {
public:int sum = 0;void traversal(TreeNode* root, int& sum){if(root == nullptr) return;traversal(root->right, sum);sum += root->val;root->val = sum;traversal(root->left, sum);}TreeNode* convertBST(TreeNode* root) {traversal(root, sum);return root;}
};