将有序数组转换为二叉搜索树
描述
简单
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
解题
使用二分的方法建树,这样就不用考虑平衡
的问题
当然使用不同的方法建树,最后的结果是不唯一的
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
return [] if len(nums)==0 else self.build(nums, 0, len(nums)-1)
def build(self, nums, left, right):
if left > right:
return None
mid = (left + right) // 2
root = TreeNode(nums[mid])
root.left = self.build(nums, left, mid-1)
root.right = self.build(nums, mid+1, right)
return root
有序链表转换二叉搜索树
描述
中等
给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9],
一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
解题
可以按照上题,先扫描链表,将数据保存至list列表中,然后二分建树
按照链表做的话,其主要思路就是找到链表的中点
设置两个指针,fast和slow,fast扫描一次后移2个位置,slow扫描一次后移1个位置
当fast到达链尾时,slow为链表中间的位置
然后就可以二分建树了
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def sortedListToBST(self, head: ListNode) -> TreeNode:
return self.build(head, None)
def build(self, head, tail):
if head == tail:
return None
fast = head
slow = head
while fast != tail and fast.next != tail:
fast = fast.next.next
slow = slow.next
root = TreeNode(slow.val)
root.left = self.build(head, slow)
root.right = self.build(slow.next, tail)
return root