难度:中等
题目描述:
给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9],
一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
解题思路:
遍历链表将值存入数组,递归构建每个子树。将数组从最中间项分割得到三个部分:子数组1,中间项,子数组2。将中间项作为当前节点的val,对子数组1和子数组2分别递归执行原方法,得到的两个子树分别作为上一个节点的左子树与右子树
var sortedListToBST = function(head) {
let cur = head;
let arr = [];
while (cur) {
arr.push(cur.val);
cur = cur.next;
}
function BTS(data) {
if (!data.length) return null;
const root = new TreeNode(null);
let min = Math.floor(data.length / 2);
root.left = BTS(data.slice(0, min));
root.val = data[min];
root.right = BTS(data.slice(min + 1));
return root;
}
return BTS(arr);
};