2023.4.6 修剪二叉搜索树
思路:遍历而叉搜索树,如果当前节点值等于右边接,就删除其右子树,如果当前节点大于了右边界,删除起右子树,同时链接其左子树,继续遍历左子树。
如果当前节点值等于左边界,删除其左子树,如果当前节点值小于左边界,删除左子树后,继续遍历右子树。
class Solution {
public TreeNode trimBST(TreeNode root, int low, int high) {
if(root.left == null && root.right == null && (root.val <low || root.val > high)){
return null;
}
return delete(root,low,high);
}
public TreeNode delete(TreeNode root,int low,int high){
//如果当前节点为null,返回null
if(root == null){
return null;
}
if(root.val>low && root.val < high){//如果当前节点值处在low和high之间,继续遍历其左右子节点
root.left = delete(root.left,low,high);
root.right = delete(root.right,low,high);
}
if(root.val == high){//当前节点值等于右边界,直接删除其右子树即可
root.right = null;
root.left = delete(root.left,low,high);
}
if(root.val == low){//当前节点值等于左边界,直接删除其左子树
root.left = null;
root.right = delete(root.right,low,high);
}
if(root.val > high){//如果遍历到当前节点值大于了右边界,此时我们需要删除当前节点及其右子树,同时还需要检查其左子树是否符合要求。
root.right = null;
root = delete(root.left,low,high);//检查其左子树节点
}
if(root.val < low){//如果遍历到当前节点值小于了左边界,此时我们需要删除当前节点及其左子树,同时还需要检查其右子树是否符合要求
root.left = null;
root = delete(root.right,low,high);
}
//最后返回root
return root;
}
}
2023.4.6 将有序数组转换为二叉搜索树
108. 将有序数组转换为二叉搜索树 - 力扣(LeetCode)
直接将中间的元素当做根节点?,然后从左侧开始依次当做左子树的节点,。这样不太行
如何才能保持二叉树的平衡?以当前数组中间的元素作为根节点,随后在两侧区间的中间元素取为左节点和右节点
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
return createTree(nums,0,nums.length);
}
public TreeNode createTree(int[] nums,int left,int right){
if(left >= right){//范围越界,结束判断
return null;
}
int mid = (right + left) / 2;
TreeNode root = new TreeNode(nums[mid]);
root.left = createTree(nums,left,mid);
root.right = createTree(nums,mid+1,right);
return root;
}
}
2023.4.7 把二叉搜索树转换为累加树
538. 把二叉搜索树转换为累加树 - 力扣(LeetCode)
实际上,这个赋值的顺序就是先遍历右子节点,遍历到最后一个叶子结点后,然后赋值,再遍历其左子节点。这个遍历顺序就是中序遍历的反向版。因此我们可以通过递归或者迭代来实现。
class Solution {
int sum = 0;
public TreeNode convertBST(TreeNode root) {
inorder(root);
return root;
}
public void inorder(TreeNode root){
if(root == null){
return;
}
inorder(root.right);
//执行逻辑,记录当前节点值
sum += root.val;
root.val = sum;
inorder(root.left);
}
}