leetcode: 109. Convert Sorted List to Binary Search Tree
阿新 • • 發佈:2018-11-08
Difficulty
Medium.
Problem
Given a singly linked list where elements are sorted in ascending order, convert it to a height balanced BST.
For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1 .
Example:
Given the sorted linked list: [-10,-3,0,5,9],
One possible answer is: [0,-3,9,-10,null,5], which represents the following height balanced BST:
0
/ \
-3 9
/ /
-10 5
AC
class Solution():
def sortedListToBST(self, head):
if not head:
return None
if not head.next:
root, root.left, root.right = TreeNode(head.val), None, None
return root
preslow = slow = fast = head
while fast and fast.next:
preslow, slow, fast = slow, slow.next, fast.next.next
preslow.next = None
root = TreeNode(slow.val)
root.left = self.sortedListToBST(head)
root.right = self.sortedListToBST(slow.next)
return root