给你一棵二叉搜索树,请你返回一棵 平衡后 的二叉搜索树,新生成的树应该与原来的树有着相同的节点值。
如果一棵二叉搜索树中,每个节点的两棵子树高度差不超过 1 ,我们就称这棵二叉搜索树是 平衡的 。
如果有多种构造方法,请你返回任意一种。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/balance-a-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
直接手撕hold不住啊,中序,重新建树。
import java.util.ArrayList;
import java.util.List;
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
class Solution {
public TreeNode balanceBST(TreeNode root) {
List<TreeNode> list = new ArrayList<>();
inOrder(root, list);
TreeNode[] treeNodes = list.toArray(new TreeNode[list.size()]);
return buildTree(treeNodes, 0, treeNodes.length - 1);
}
private TreeNode buildTree(TreeNode[] treeNodes, int start, int end) {
if (start <= end) {
int mid = (start + end) / 2;
TreeNode treeNodeRoot = treeNodes[mid];
treeNodeRoot.left = buildTree(treeNodes, start, mid - 1);
treeNodeRoot.right = buildTree(treeNodes, mid + 1, end);
return treeNodeRoot;
}
return null;
}
private void inOrder(TreeNode root, List<TreeNode> list) {
if (root != null) {
inOrder(root.left, list);
list.add(root);
inOrder(root.right, list);
}
}
}