题目
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
要点
首先你应该要能够理解二叉搜索树的基本性质,其在中序遍历的时候刚好是一个从小到大排序的数组。
我们举一个简单的例子来看 [1,2,3] 这个数组的构成的树应当是
你应该能发现一个有趣的现象,就是我们把2这个数往上提就构成的一个二叉搜索树 ,此时1跟3分别在2的左边跟右边。数组中间的数字是某子树的根节点。
将有序数组转换为二叉搜索树动画演示
递归解法
代码片段
func sortedArrayToBST(nums []int) *TreeNode {
if len(nums) == 0 {
return nil
}
var t = &TreeNode{}
sortedArrayToBSTNode(nums, t)
return t
}
func sortedArrayToBSTNode(nums []int, node *TreeNode) {
//part1
mid := len(nums) / 2
node.Val = nums[mid] //取出这个数组中间的数字当做的根结点
leftNums := nums[0:mid] //将数组分割成左右两边,注意边界不包含mid这个位置的数。
RightNums := nums[mid+1:]
//part2
if len(leftNums) > 0 {
node.Left = &TreeNode{}
sortedArrayToBSTNode(leftNums, node.Left)
}
//part3
if len(RightNums) > 0 {
node.Right = &TreeNode{}
sortedArrayToBSTNode(RightNums, node.Right)
}
}
代码解释
其实这里是一个先序遍历的过程,我们可以从动画中清晰的看到这颗树构成的过程。
代码里面的注释part1就是在创建根结点的过程,而part2,part3就是结点旁边的两个连接线。