给你一棵二叉搜索树，请你返回一棵 平衡后 的二叉搜索树，新生成的树应该与原来的树有着相同的节点值。如果有多种构造方法，请你返回任意一种。

如果一棵二叉搜索树中，每个节点的两棵子树高度差不超过 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] 也是一个可行的构造方案。

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {TreeNode}
 */
var balanceBST = function (root) {
	let arr = centerTraverse(root);
    console.log(arr, 'arr')

	function gen(nums) {
		if (!nums.length) return null;

		let m = Math.floor((nums.length - 1) / 2);
		let head = new TreeNode(nums[m]);

		head.left = gen(nums.slice(0, m));
		head.right = gen(nums.slice(m + 1));

    return head
	}

  return gen(arr)
};

var centerTraverse = function (root) {
	let ans = [];

	function helper(root) {
		if (root === null) return;
		helper(root.left);
		ans.push(root.val);
		helper(root.right);
	}

	helper(root);

	return ans;
};

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/balance-a-binary-search-tree