669. 修剪二叉搜索树
可以用递归的方式解决。这个问题的关键是要找到数组的中间元素作为二叉搜索树的根。然后将数组的左半部分和右半部分递归地转换为左右子树。这样,由于左右子树的元素数量相等或者相差1,可以保证得到的二叉搜索树是高度平衡的
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
// 对空数组或null进行判断
if (nums == null || nums.length == 0) {
return null;
}
// 调用递归方法来构建二叉搜索树
return buildBST(nums, 0, nums.length - 1);
}
private TreeNode buildBST(int[] nums, int left, int right) {
// 如果左索引大于右索引,则返回null,表示当前这部分数组不需要构建二叉搜索树
if (left > right) {
return null;
}
// 选择中间元素作为二叉搜索树的根节点,这样可以保证二叉搜索树是平衡的
int mid = left + (right - left) / 2;
TreeNode node = new TreeNode(nums[mid]);
// 递归地构建左子树
node.left = buildBST(nums, left, mid - 1);
// 递归地构建右子树
node.right = buildBST(nums, mid + 1, right);
// 返回构建的二叉搜索树的根节点
return node;
}
}
538.把二叉搜索树转换为累加树
可以用二叉搜索树的反向中序遍历(右-根-左)来解决。反向中序遍历二叉搜索树可以得到一个递减的序列。遍历时,用一个变量sum来存储已经遍历过的节点的值的总和,然后把这个总和赋值给当前节点。
class Solution {
// 初始化 sum 为 0
int sum = 0;
public TreeNode convertBST(TreeNode root) {
// 如果树为空,直接返回 null
if (root == null) {
return null;
}
// 首先遍历右子树
convertBST(root.right);
// 将当前节点的值加到 sum 中
sum += root.val;
// 将 sum 赋值给当前节点
root.val = sum;
// 然后遍历左子树
convertBST(root.left);
// 返回修改后的根节点
return root;
}
}
在这段代码中,变量sum
用于存储已经遍历过的节点的值的总和。函数convertBST
首先遍历右子树,然后处理根节点,最后遍历左子树,这样就实现了反向中序遍历。在处理根节点时,把sum
赋值给当前节点,然后更新sum
的值。最后返回修改后的根节点。
什么是累加树?
累加树是一种特殊的二叉树,其中每个节点的新值等于原树中大于或等于该节点值的所有值之和。
例如,给定一个如下的二叉搜索树:
5
/ \
2 13
转换后,树的形状保持不变,只是节点值变为原来大于等于该节点值的所有节点值之和:
18
/ \
20 13
转换规则是这样的:根据二叉搜索树的特性,一个节点的值等于它的右子树的所有节点值之和加上它自己的值和它左子树中所有比它大的节点的值之和。所以,如果我们用逆中序遍历(右-根-左的顺序)来遍历这棵树,那么我们可以使用一个累加器(accumulator)来记录已经累加过的节点值,然后在遍历到一个新节点时,把累加器的值加到新节点的值中。这样,我们就可以在遍历的过程中完成节点值的转换,最终得到一个累加树。