Leetcode 669 修剪二叉搜索树
题目链接 669 修剪二叉搜索树
本题上来看还是有难度的,也算是对删除二叉树节点的升级版,但是想清楚还是比较简单的
下面上代码加注释:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int low, int high) {
if(root == NULL){
return NULL;
}
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->right = trimBST(root->right,low,high);
return root;
}
};
Leetcode 108. 将有序数组转换为二叉搜索树
题目链接 108 将有序数组转换为二叉搜索树
本题目和前面的中序后序数组构造二叉树的基本思路是一样的,确定中间节点之后,再去构造左右子树,在通过递归来构造左右子树的左右子树。
本题目是有序数组,简单一些,找到中间值,确定好区间闭开情况(变量唯一性)
下面上代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
private:
TreeNode* traversal(vector<int>& nums,int left,int right){//这里的&是引用的作用,可以减少内存
if(left>right){//这里默认左闭右闭
return NULL;
}
int mid = (left+right)/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;
}
};
Leetcode 538. 把二叉搜索树转换为累加树
题目链接 538 把二叉搜索树转换为累加树
本题目猛的一看没有看懂,最后还是看来卡哥的视频才看懂的,原来是这个意思,我觉得双指针是能想到的,但是要主要在转化过程中是倒叙数值相加的,我们就需要右中左的遍历,下面上代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
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;
}
};
后有总结 end