669. 修剪二叉搜索树
1、代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public TreeNode trimBST(TreeNode root, int low, int high) {
if(root == null) return null;
if(root.val < low){
//通过返回值的形式进行修剪
//根据二叉搜索树的特性2
return trimBST(root.right,low,high);
}
if(root.val > high){
return trimBST(root.left,low,high);
}
root.left = trimBST(root.left,low,high);
root.right = trimBST(root.right,low,high);
return root;
}
}
2、分析
(1)本题充分利用了二叉搜索树的特性:
1)
if(root.val < low){
return trimBST(root.right,low,high);
}
这里判断当根节点的值小于题目规定的最小值时,函数返回根节点的右节点并进行递归,这是由于在二叉搜索树的规则中,根节点的左子树的值全部小于根节点。所以这里通过判断根节点的值之后可以直接删除左子树。
2)
root.left = trimBST(root.left,low,high);
root.right = trimBST(root.right,low,high);
这里通过一左一右两个指针遍历二叉搜索树的节点,对节点一一判断,左右指针互不干扰。
(2)删除根节点的小技巧就是: 对于二叉搜索树的节点删除可以通过函数的返回值改变指向,直接删除根节点。
108.将有序数组转换为二叉搜索树
1、代码1(BC)
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
int left = 0;
int right = nums.length - 1;
int mid = left + (right - left)/2;
TreeNode node = new TreeNode(nums[mid]);
TreeNode leftPre = node;
TreeNode rightPre = node;
while(left < mid){
leftPre.left = new TreeNode(nums[mid - 1 - left]);
leftPre = node.left;
System.out.println("nums[mid - 1]: " + nums[mid - 1 - left] );
System.out.println("left: " + left);
left++;
}
while(mid < right){
rightPre.right = new TreeNode(nums[mid + 1]);
rightPre = rightPre.right;
mid++;
}
return node;
}
}
2、分析
(1)构建二叉树成功,但不满足高度平衡,主要是对于“高度平衡”的定义理解不当。
3、代码2
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
//左闭右闭
return traversal(nums,0,nums.length - 1);
}
public TreeNode traversal(int[] nums,int left,int right){
if(left > right){
return null;
}
int mid = left + (right - left)/2;
TreeNode node = new TreeNode(nums[mid]);
node.left = traversal(nums,left,mid - 1);
node.right = traversal(nums,mid + 1,right);
return node;
}
}
4、分析
(1)本题是将数组转化为二叉搜索树,用左闭右闭的思路更容易操作索引。
(2)由于本题需要的二叉搜索树需要高度平衡,所以我们只能将每个根节点及其左右子树用对半分的思路将其切割,因为对半切割之后每个数组的长度都是一致的,换算成二叉树的话就是高度差的绝对值不会大于1。
(3)对半切割之后需要明确递归函数的参数,即数组的左闭右闭区间。对半切割一定长度后,当区间的左右分界大小不符合递增顺序时,说明数组此时为空,二叉树返回null。当区间左右分界大小相等时,说明数组范围内只剩下一个元素,由于此时mid的位置就在该分界位置,所以直接建立一个节点并将节点返回。综上:对半切割建立好全部的mid节点之后(完成不重复和越界(left、right)的全部情况),当左右指针在同一个位置时将该节点建立,当左右指针位置相反时返回null。
538.把二叉搜索树转换为累加树
1、代码(AC)
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public int sum = 0;
public TreeNode convertBST(TreeNode root) {
if(root == null){
return null;
}
//右中左的顺序进行 遍历
convertBST(root.right);
root.val = sum + root.val;
sum = root.val;
convertBST(root.left);
return root;
}
}
2、分析
(1)根据题目描述,可以明显看出这是右中左的遍历方式,所以在代码中进行稍微调节即可。