669. 修剪二叉搜索树
题目
给定一个二叉搜索树,同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树,使得所有节点的值在[L, R]中 (R>=L)
文章
视频
只要节点小于low,节点直接替换为节点右子树。
返回root
还挺简单的
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int low, int high) {
if(root==NULL) return NULL;
if(root->val<low)
{
root=trimBST(root->right,low,high);
}
else if(root->val>high)
{
root=trimBST(root->left,low,high);
}
else
{
root->left=trimBST(root->left,low,high);
root->right=trimBST(root->right,low,high);
}
return root;
}
};
迭代法没想出来
分为三步:
1 将root移动到[L, R] 范围内,注意是左闭右闭区间
2 剪枝左子树
3 剪枝右子树
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int L, int R) {
if (!root) return nullptr;
// 处理头结点,让root移动到[L, R] 范围内,注意是左闭右闭
while (root != nullptr && (root->val < L || root->val > R)) {
if (root->val < L) root = root->right; // 小于L往右走
else root = root->left; // 大于R往左走
}
TreeNode *cur = root;
// 此时root已经在[L, R] 范围内,处理左孩子元素小于L的情况
while (cur != nullptr) {
while (cur->left && cur->left->val < L) {
cur->left = cur->left->right;
}
cur = cur->left;
}
cur = root;
// 此时root已经在[L, R] 范围内,处理右孩子大于R的情况
while (cur != nullptr) {
while (cur->right && cur->right->val > R) {
cur->right = cur->right->left;
}
cur = cur->right;
}
return root;
}
};
108.将有序数组转换为二叉搜索树
题目
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1
文章
视频
挺简单的。找中间节点,递归处理左右子树
class Solution {
public:
TreeNode* sort(vector<int>& nums, int min, int max) {
if (min <= max) {
int m = (min + max + 1) / 2;
TreeNode* root = new TreeNode(nums[m]);
root->left = sort(nums, min, m - 1);
root->right = sort(nums, m + 1, max);
return root;
}
else
return NULL;
}
TreeNode* sortedArrayToBST(vector<int>& nums) {
return sort(nums, 0, nums.size()-1); //左闭右闭 别搞错了
}
};
迭代法有点复杂,想不出来。
通过三个队列来模拟,一个队列放遍历的节点,一个队列放左区间下标,一个队列放右区间下标。一层一层构造
class Solution {
public:
TreeNode* sortedArrayToBST(vector<int>& nums) {
if (nums.size() == 0) return nullptr;
TreeNode* root = new TreeNode(0); // 初始根节点
queue<TreeNode*> nodeQue; // 放遍历的节点
queue<int> leftQue; // 保存左区间下标
queue<int> rightQue; // 保存右区间下标
nodeQue.push(root); // 根节点入队列
leftQue.push(0); // 0为左区间下标初始位置
rightQue.push(nums.size() - 1); // nums.size() - 1为右区间下标初始位置
while (!nodeQue.empty()) {
TreeNode* curNode = nodeQue.front();
nodeQue.pop();
int left = leftQue.front(); leftQue.pop();
int right = rightQue.front(); rightQue.pop();
int mid = left + ((right - left) / 2);
curNode->val = nums[mid]; // 将mid对应的元素给中间节点
if (left <= mid - 1) { // 处理左区间
curNode->left = new TreeNode(0);
nodeQue.push(curNode->left);
leftQue.push(left);
rightQue.push(mid - 1);
}
if (right >= mid + 1) { // 处理右区间
curNode->right = new TreeNode(0);
nodeQue.push(curNode->right);
leftQue.push(mid + 1);
rightQue.push(right);
}
}
return root;
}
};
538.把二叉搜索树转换为累加树
题目
累加树,书中每个节点=大于等于当前节点的值做累加
文章讲解
视频
没想到双指针法,因为对二叉树遍历的本质不熟悉:中序就是递增
二叉树中序遍历是递增,反中序遍历(右中左)是递减。
如何想到双指针法:二叉树提取为数组,一个指向当前,一个指向上一个大于自己且已完成累加操作的节点
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;
}
};
二叉树总结
对题目进行分类汇总以及总结。自己复习完在过来看