参考文献链接:代码随想录
本人代码是Java版本的,如有别的版本需要请上代码随想录网站查看。
669. 修剪二叉搜索树
解题思路
这道题目是判断范围,利用二叉搜素树的特性遍历即可,比如头节点满足题意,那么就左右各自递归去寻找不满足的,当找到不满足的时,要判断是大于high还是小于low,根据不同情况对该节点的左或者右子树进行全裁。具体情况我们代码说话。
代码示例
class Solution {
public TreeNode trimBST(TreeNode root, int low, int high) {
if(root == null){
return null;
}
//如果小于low,说明它的左子树全小于,直接都不要了。
//为什么再次递归呢,因为你的右子树上来后可能还是不满足题意。
if(root.val < low){
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;
}
}
108.将有序数组转换为二叉搜索树
解题思路
这道题目我们先抓题眼:平衡树,搜索树,有序数组。
首先搜索树和有序数组对的上,我们只需要把数组中间的数当成头节点,然后把数组拆分到它的左右子树即可,递归法。
同时其实我们也完成了平衡树的要求,因为我们是从中分开依次分给左右,所以一定是平衡树。
代码示例
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
if(nums.length == 1){
return new TreeNode(nums[0]);
}
if(nums.length == 0){
return null;
}
TreeNode root = new TreeNode(nums[nums.length / 2]);
root.left = sortedArrayToBST(Arrays.copyOfRange(nums,0,nums.length / 2));
root.right = sortedArrayToBST(Arrays.copyOfRange(nums,nums.length / 2 + 1,nums.length));
return root;
}
}
538.把二叉搜索树转换为累加树
解题思路
首先要先通过题目示例明白累加树是什么,其实也就是右中左的顺序去累加即可,那么我们只需要递归去右中左遍历,并且有个全局变量sum一直累加并且给节点赋值即可。
代码示例
class Solution {
int sum;
public TreeNode convertBST(TreeNode root) {
sum = 0;
covert(root);
return root;
}
public void covert(TreeNode node){
if(node == null){
return;
}
covert(node.right);
sum += node.val;
node.val = sum;
covert(node.left);
}
}