题目:
**将一个按照升序排列的有序数组、链表,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组、链表: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
… 0
. /. …\
-3 … 9
. / … /
-10 . 5
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。**
思路:
采用递归来解决问题,如果以数组、链表最中间的值来作为根节点,那么左、右子树的节点数量差不超过1,那么中间值作为分界点,用前面的值重建左子树,后面的值重建右子树,递归即可解决;
该题的关键就在于找中间点,数组只需要通过下标计算即可;链表则可以通过快慢指针来找,但时间复杂度将再原有的基础上提高logN,可以考虑先将链表转换成数组,即可优化掉这部分时间复杂度;
再考虑将链表看作是 二叉搜索树的中序遍历结果,然后根据中序遍历的过程重建二叉树,这将不再需要将链表转换成数组,可优化掉这部分空间复杂度,但比较难以理解,具体看链表转二叉树代码;
代码分析:
- 代码较简单,直接看代码;根据中序遍历重建的过程只能靠自己理解了;
代码实现:
数组转二叉树:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
if(nums == null) return null;
return rebuild(nums,0,nums.length-1);
}
private TreeNode rebuild(int[] nums,int start,int end){
if(start > end) return null;
int center = (start + end)/2;
TreeNode node = new TreeNode(nums[center]);
node.left = rebuild(nums,start,center-1);
node.right = rebuild(nums,center+1,end);
return node;
}
}
链表转二叉树:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
ListNode head;
public TreeNode sortedListToBST(ListNode head) {
this.head = head;
int i = 0;
ListNode node = head;
while(node != null){
++i;
node = node.next;
}
return rebuild(0,i-1);
}
public TreeNode rebuild(int left,int right){
if(left > right)return null;
int center= (left + right) / 2;
TreeNode leftNode = rebuild(left,center-1);
TreeNode root = new TreeNode(head.val);
root.left = leftNode;
head = head.next;
root.right = rebuild(center + 1,right);
return root;
}
}