修剪二叉搜索树
题目:给你二叉搜索树的根节点 root ,同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树,使得所有节点的值在[low, high]中。修剪树 不应该 改变保留在树中的元素的相对结构 (即,如果没有被移除,原有的父代子代关系都应当保留)。 可以证明,存在 唯一的答案 。
所以结果应当返回修剪好的二叉搜索树的新的根节点。注意,根节点可能会根据给定的边界发生改变。
思路:我们要怎么删除一个二叉树的节点,我们都是通过向上返回一个节点然后上层递归函数中的节点的左右孩子接受到这个返回的节点,从而删除我们想要删除的节点;是在想不明白就找个二叉搜索树来手动模拟一下过程就清楚了
class Solution {
public TreeNode trimBST(TreeNode root, int low, int high) {
if(root == null){// 终止条件
return null;
}
if(root.val < low){
// 因为是二叉搜索树,所有当前节点的左子树肯定是不符合要求的,这里就不处理,下面会往左遍历(代码1),然后就会到上面的终止条件,然后返回null,然后当前节点的左孩子就会接收到一个null,这样就删除了左子树
// 但是右子树有可能存在不满足条件的,所以需要在一次修剪,这里返回的节点就是满足要求的子树的更节点,返回 的节点会在(代码1处接收到)
return trimBST(root.right,low,high);
}
if(root.val > high){
// 同上
return trimBST(root.left,low,high);
}
// 代码1
root.left = trimBST(root.left,low,high);
root.right = trimBST(root.right,low,high);
return root;
}
}
将有序数组转换成二叉搜索树
题目:给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
思路:构造二叉树,这里可以用两个变量标记构造哪个区间的二叉树,我们统一采用左闭右开
的规则
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
return traversal(nums,0,nums.length);
}
// 构造二叉树 区间采用左闭右开
private TreeNode traversal(int[] nums,int left,int right){
if(left >= right){// 终止条件
return null;
}
// 中间的值作为根节点
int mid = (left+right) / 2;
TreeNode node = new TreeNode(nums[mid]);
// 构造左子树
node.left = traversal(nums,left,mid);
// 构造右子树
node.right = traversal(nums,mid + 1,right);
return node;
}
}
把二叉搜索树转换为累加树
题目:给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
提醒一下,二叉搜索树满足下列约束条件:
节点的左子树仅包含键 小于 节点键的节点。
节点的右子树仅包含键 大于 节点键的节点。
左右子树也必须是二叉搜索树。
思路:由题目可知:我们是需要从大的值开始,已经往前累加,所有我们采用的遍历顺序就是右中左
class Solution {
// 记录前一个节点的值
TreeNode pre = null;
public TreeNode convertBST(TreeNode root) {
if(root == null){// 终止条件
return null;
}
// 右
root.right = convertBST(root.right);
// 中
if(pre != null){
root.val = root.val + pre.val;
}
pre = root;
// 左
root.left = convertBST(root.left);
return root;
}
}