#Java #二叉树
Feeling and experiences:
给你二叉搜索树的根节点 root
,同时给定最小边界low
和最大边界 high
。通过修剪二叉搜索树,使得所有节点的值在[low, high]
中。修剪树 不应该?改变保留在树中的元素的相对结构 (即,如果没有被移除,原有的父代子代关系都应当保留)。 可以证明,存在?唯一的答案?。
所以结果应当返回修剪好的二叉搜索树的新的根节点。
注意,根节点可能会根据给定的边界发生改变。
/**
* 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;
}
// 如果当前节点的值小于low,修剪左子树并递归右子树
if (root.val < low) {
return trimBST(root.right, low, high);
}
// 如果当前节点的值大于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;
}
}
1. 递归的核心:
? 针对给定的根节点,以及指定的范围?low?和?high,递归地修剪树,确保所有节点的值都在这个范围内。
2. 处理当前节点:
? 如果当前节点的值小于?low,那么它的左子树所有节点的值也一定小于?low。因此,保留修剪过的右子树。
? 如果当前节点的值大于?high,那么它的右子树所有节点的值也一定大于?high。因此,保留修剪过的左子树。
? 如果当前节点的值在?[low,?high]?范围内,递归地修剪左右子树。
3. 递归终止条件:
? 当遇到?null?节点时,递归终止。?
和昨天写的 删除二叉搜索树的节点? 比较一下:
给你一个整数数组 nums
,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 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 sortedArrayToBST(int[] nums) {
//该数组已经是升序排列的了,按照二叉搜索树的性质进行即可
//还要保证其是一颗平衡二叉树
return create(nums,0,nums.length - 1);
}
public TreeNode create(int[] nums,int left, int right){
if(left > right){
return null;
}
//选择中间的为root
int mid = (left + right) >> 1;
TreeNode root = new TreeNode(nums[mid]);
root.left = create(nums,left,mid - 1);
root.right = create(nums,mid+1,right);
return root;
}
}
?这道题很简单,构造一棵树写过很多次了,但是多了一个要求 :树为二叉平衡树
只要我们选择中间的数为根,那么在递归的时候,就能尽可能保证平衡左右子树的高度。
给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node
?的新值等于原树中大于或等于?node.val
?的值之和。
提醒一下,二叉搜索树满足下列约束条件:
/**
* 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 convertBST(TreeNode root) {
if(root == null) {
return root;
}
//利用逆中序的方法
Stack<TreeNode> stack = new Stack<TreeNode>();
dfs(root, stack);
TreeNode node = stack.pop();
while(!stack.isEmpty()) {
TreeNode cur = stack.pop();
cur.val += node.val;
node = cur;
}
return root;
}
private void dfs(TreeNode root, Stack<TreeNode> stack) {
if(root != null) {
dfs(root.left, stack);
stack.push(root);
dfs(root.right, stack);
}
}
}
?
?
?以下是递归遍历:
/**
* 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 {
int sum = 0;
public TreeNode convertBST(TreeNode root) {
if (root != null) {
convertBST(root.right);
sum += root.val;
root.val = sum;
convertBST(root.left);
}
return root;
}
}
该题目,主要就是用到的中序的逆序,读懂题目就简单了!
盛气光引炉烟,
素草寒生玉佩。
Fighting!
?