108. 将有序数组转换为二叉搜索树
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/
- 3 9
/ /
-10 5
分析:
二叉搜索树的中序遍历结果就是有序的序列。而构造一颗唯一地二叉树需要中序序列,还需要先序或后序结果。
本题给出了另一个条件,高度平衡的二叉搜索树,意味着任意结点两端的子树高度相差不超过1。那么只需要递归时将序列中间值作为根结点即可。
代码(Golang):
func sortedArrayToBST(nums []int) *TreeNode {
var construct func(left int, right int) *TreeNode
construct = func(left int, right int) *TreeNode {
if left > right {
return nil
}
mid := (left + right) / 2
root := TreeNode{Val: nums[mid]}
root.Left = construct(left, mid - 1)
root.Right = construct(mid + 1, right)
return &root
}
return construct(0, len(nums) - 1)
}