669.修剪二叉搜索树
讲解链接:代码随想录-669.修剪二叉搜索树
提示:使用后序遍历。判断当前节点是否在区间内,不在的话,把筛选后符合条件的子节点给赋值就行。
public TreeNode trimBST(TreeNode root, int low, int high) {
if (root == null) return null;
// 后序遍历,从最下层开始修剪。
root.left = trimBST(root.left, low, high);
root.right = trimBST(root.right, low, high);
// 如果节点值小于最小区间,返回可右节点。因为是从最下层开始修剪,所以可以保证当前下层修剪后的值都是符合要求的。所以直接返回右节点就好了
if (root.val < low) {
return root.right;
}
// 如果节点值大于最大区间,返回左节点。同上
if (root.val > high) {
return root.left;
}
// 在区间内,直接返回就好了。
return root;
}
108.将有序数组转换为二叉搜索树
构造二叉树一般都是用前序遍历(中左右),首先构建中节点,然后左右子节点。有序数组,构造搜索二叉树。
中间点毫无疑问就是数组中间那个数,左右子节点就是被隔断成两个数组后,各自中间的数字。以此类推就很好理解了。
涉及到数组分割,编码的时候注意好左闭右开的判断。
public TreeNode sortedArrayToBST(int[] nums) {
return sortedArrayToBST(nums, 0, nums.length);
}
public TreeNode sortedArrayToBST(int[] nums, int left, int right) {
if (left >= right) {
return null;
}
// int mid = (left + right) / 2; 这里劲量不要用这种方式,因为如果 left 和 right 值很大,相加会造成越界,结果就不对了。
int mid = left + (right - left) / 2;
TreeNode root = new TreeNode(nums[mid]);
root.left = sortedArrayToBST(nums, left, mid);
root.right = sortedArrayToBST(nums, mid + 1, right);
return root;
}
538.把二叉搜索树转换为累加树
这道题很简单,标准的搜索二叉树解法,不过用到的是翻转的中序遍历:右中左。
TreeNode preNode;
public TreeNode convertBST(TreeNode root) {
if (root == null) return root;
convertBST(root.right);
if (preNode != null) {
root.val = root.val + preNode.val;
}
preNode = root;
convertBST(root.left);
return root;
}
总结
今天额外的轻松。