二叉树9|669. 修剪二叉搜索树|108.将有序数组转换为二叉搜索树|538.把二叉搜索树转换为累加树
一、669. 修剪二叉搜索树
题目连接:669. 修剪二叉搜索树 - 力扣(LeetCode)
- 思路:用递归法,递归返回本子树的根节点。如果root(当前节点)的元素小于low的数值,那么应该递归右子树,并返回右子树符合条件的头结点。 如果root(当前节点)的元素大于high的,那么应该递归左子树,并返回左子树符合条件的头结点。
- 注意:这题与插入和删除指定节点有点不同,这题删除的是很多个节点,不能让本节点的父节点的左右指针指向本节点的孩子节点,因为孩子节点可能也是要删除的节点。只能采用遍历子树,保留子树合法的节点后,返回子树的根节点。
class Solution {
public TreeNode trimBST(TreeNode root, int low, int high) {
if(root == null) return root;
if(root.val < low){
TreeNode right = trimBST(root.right, low, high);
return right;
}
if(root.val > high){
TreeNode left = trimBST(root.left, low, high);
return left;
}
//删除节点逻辑
root.left = trimBST(root.left, low, high);
root.right = trimBST(root.right, low, high);
return root;
}
}
二、108.将有序数组转换为二叉搜索树
题目连接:108. 将有序数组转换为二叉搜索树 - 力扣(LeetCode)
给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
- 注意循环不变量规则,这里我采用的是左闭右开。
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
return traversal(nums, 0, nums.length);
}
public TreeNode traversal(int[] nums, int left, int right){
if(left >= right) return null;
if(right - left == 1) return new TreeNode(nums[left]);
int mid = left + (right - left) / 2;
TreeNode treeNode = new TreeNode(nums[mid]);
treeNode.left = traversal(nums,left,mid);
treeNode.right = traversal(nums,mid +1,right);
return treeNode;
}
}
三、538.把二叉搜索树转换为累加树
题目连接:538. 把二叉搜索树转换为累加树 - 力扣(LeetCode)
- 可以用双指针法,用 pre 记录前一个节点的值,采用倒中序遍历,即右中左的顺序遍历二叉树,进行累加操作。
class Solution {
int pre = 0;
public TreeNode convertBST(TreeNode root) {
traversal(root);
return root;
}
public void traversal(TreeNode root){
if(root == null) return;
traversal(root.right);//右
root.val += pre;
pre = root.val;
traversal(root.left);//左
}
}