给你一棵二叉搜索树,请你返回一棵 平衡后 的二叉搜索树,新生成的树应该与原来的树有着相同的节点值。如果有多种构造方法,请你返回任意一种。
如果一棵二叉搜索树中,每个节点的两棵子树高度差不超过 1 ,我们就称这棵二叉搜索树是 平衡的 。
示例 1:
输入:root = [1,null,2,null,3,null,4,null,null]
输出:[2,1,3,null,null,null,4]
解释:这不是唯一的正确答案,[3,1,4,null,2,null,null] 也是一个可行的构造方案。
示例 2:
输入: root = [2,1,3]
输出: [2,1,3]
提示:
树节点的数目在 [1, 104] 范围内。
1 <= Node.val <= 105
/**
* 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 {
List<Integer> list = new ArrayList<>();
public TreeNode balanceBST(TreeNode root) {
get(root);
return dfs(0, list.size() - 1);
}
void get(TreeNode root) {
if (root == null) return;
get(root.left);
list.add(root.val);
get(root.right);
}
TreeNode dfs(int l, int r) {
if (l > r) return null;
int mid = l + r >> 1;
TreeNode root = new TreeNode(list.get(mid));
root.left = dfs(l, mid - 1);
root.right = dfs(mid + 1, r);
return root;
}
}