1.题目描述
给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9],
一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
在真实的面试中遇到过这道题?
2.解题思路
这道题和leetcode108.将有序数组转换为二叉搜索树很相似。
如果将二叉搜索树按中序遍历的话,得到的就是一个有序数组了。那么反过来,我们可以得知,根节点应该是有序链表的中间点,从中间点分开为左右两个有序链表,在分别找出其中间点作为原中间点的左右两个子节点,递归二分查找的过程。
链表的中点可以用快慢指针实现。慢指针走一步,快指针走两步,快指针走到终点时,慢指针指向的是中点。
3.代码实现
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, x):
# self.val = x
# self.next = None
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def sortedListToBST(self, head):
"""
:type head: ListNode
:rtype: TreeNode
"""
if not head:
return
if not head.next:
return TreeNode(head.val)
slow = head
fast = head
last = head
while fast.next and fast.next.next:
last = slow
slow = slow.next
fast = fast.next.next
# 这里要注意赋值的顺序
head2 = slow.next
last.next = None
root = TreeNode(slow.val)
# 当对两个节点递归时,左节点本应该为空,如果不加这句,左节点会被赋值为根节点的值
if head != slow:
root.left = self.sortedListToBST(head)
root.right = self.sortedListToBST(head2)
return root