leetcode109.有序链表转换二叉搜索树

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