给定一个单链表的头节点 head ,其中的元素 按升序排序 ,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差不超过 1。
示例 1:
输入: head = [-10,-3,0,5,9]
输出: [0,-3,9,-10,null,5]
解释: 一个可能的答案是[0,-3,9,-10,null,5],它表示所示的高度平衡的二叉搜索树。
示例 2:
输入: head = []
输出: []
提示:
head 中的节点数在[0, 2 *
1
0
4
10^4
104] 范围内
−
1
0
5
-10^5
−105 <= Node.val <=
1
0
5
10^5
105
思路
搜索二叉树要求左右子树的高度差不大于一,很容易想到,若使链表的中点作为搜索二叉树的根结点,那么左右两个树的节点数量差不大于一,如果每个子树都满足该条件,那么就可以做到左右子树的高度差不大于一。又因为该问题具有重复性,可以使用递归的方式进行解决。
代码:
public TreeNode sortedListToBST(ListNode head) {
if(head==null){
return null;
}
if(head.next == null){
return new TreeNode(head.val,null,null);
}
ListNode dummy = new ListNode(-1);
dummy.next = head;
ListNode fast = head;
ListNode slow = head;
ListNode pre = dummy;
while(fast!=null && fast.next!=null){
slow = slow.next;
fast = fast.next.next;
pre = pre.next;
}
pre.next =null;//不要忘记把前半个链表的结尾置为空
TreeNode result = new TreeNode(slow.val);
result.left = sortedListToBST(head);
result.right = sortedListToBST(slow.next);
return result;
}