题1:
指路:LeetCode669 修剪二叉搜索树
思路与代码:
清晰的题意:不在给定范围内的值删掉。与链接中的第三题相似的,我们用右子树填补被删除节点为例,使其递归调用函数,继续判断右子树中的值是否满足条件,最终返回满足条件的右子树整体,回溯给被删除节点的父节点,使其父节点直接指向其满足条件的右子树。左子树亦然。代码如下:
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int low, int high) {
if (root == NULL) return root;
if (root->val < low) {
TreeNode* right;
right = trimBST(root->right, low, high);
return right;
}
if (root->val > high) {
TreeNode* left;
left = trimBST(root->left, low, high);
return left;
}
root->left = trimBST(root->left, low, high);
root->right = trimBST(root->right, low, high);
return root;
}
};
题2:
指路:LeetCode108 将有序数组转换为二叉搜索树
思路与代码:
对于一个给定的数组,找到中间值做根节点值构造根节点;分别取左右区间;再将左右区间分别作为前面说的"给定的数组"。代码如下:
class Solution {
private:
TreeNode* traversal(vector<int>& nums, int left, int right) {
if (left > right) return NULL; // 这里我们将区间定义为左闭右闭
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;
}
};
题3:
指路:LeetCode538 把二叉搜索树转换为累加树
思路与代码:
对于这个累加树将其放在有序数组中会好理解。在有序数组中从后往前累加即可。放在二叉搜索树中也一样,因为它的特性就是中序遍历有序,所以最右边的节点一定是最大值。结合前面的思路,我们用“右中左”序列遍历,再累加即可。代码如下:
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;
}
};
二叉树的总结:
今天是二叉树的最后一篇,也许是做题少的原因,我觉得递归回溯这种东西放在二叉树里了解简直是最有效的(,当然迭代也很适合它)。从一开始的前中后序、层序遍历到路径与路径和、再然后就到构造,插入删除节点。我从一开始直接迭代到现在觉得递归才是更是它的,也算是对做题方法有了较为清晰的认知。