题目
给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9],
一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:
1 2 3 4 5
| 0 / \ -3 9 / / -10 5
|
题解
这道题类似归并排序,需要每次都把链表分为尽可能等长的两部分,然后分别对这两部分再进行生成二叉树的操作。可以用快慢指针,具体实现是:设置快慢指针各一个,慢指针步进为1,快指针步进为2,循环条件是快指针不为空且其子节点不为空。这样快指针走的长度是慢的2倍,可以保证每次循环都找到了链表的中心节点(平衡二叉树需要找中心节点)。每次循环后链表被分为: 0-premid、slow、slow.next-最后一个 这三部分。
Java代码如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
|
class Solution { public TreeNode sortedListToBST(ListNode head) { if( head==null ) return null; if( head.next==null ) return new TreeNode(head.val); ListNode slow = head; ListNode fast = head; ListNode premid = null; while( fast!=null && fast.next!=null ) { premid = slow; slow = slow.next; fast = fast.next.next; } premid.next = null; TreeNode root = new TreeNode(slow.val); root.left = sortedListToBST(head); root.right = sortedListToBST(slow.next); return root; } }
|