一个偷懒的做法就是把链表转换为数组。然后用分治法。左边递归构建子数,右边递归构建子树。。线性时间复杂度。。看上去不错。
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
Integer []nums;
public TreeNode sortedListToBST(ListNode head) {
LinkedList<Integer>temp=new LinkedList<Integer>();
while(head!=null)
{
temp.add(head.val);
head=head.next;
}
nums=new Integer[temp.size()];
temp.toArray(nums);
return sortedArrayToBST();
}
public TreeNode sortedArrayToBST() {
if(nums.length==0)
{
return null;
}
int middle=(nums.length-1)/2;
TreeNode root=new TreeNode(nums[middle]);
help(root, 0, middle-1,true);
help(root,middle+1,nums.length-1,false);
return root;
}
public void help(TreeNode root,int left,int right,boolean isLeft)
{
if(right<left)
{
return;
}
int middle=(left+right)/2;
if(isLeft)
{
root.left=new TreeNode(nums[middle]);
help(root.left,left,middle-1,true);
help(root.left,middle+1,right,false);
}
else
{
root.right=new TreeNode(nums[middle]);
help(root.right,left,middle-1,true);
help(root.right,middle+1,right,false);
}
}
}
不过还有想一想有没有优美一点的方法?
用红黑树也是一个不错的选择。但是红黑树本身实现就很复杂,对我来说当然是个复杂的数据结构。不过加上旋转也不见得效果有多么好。
看到这里有个国外的傻逼和我一个想法。先变成数组在做^_^
有人用这样的算法做
Recursive BST construction using slow-fast traversal on linked list
public TreeNode sortedListToBST(ListNode head) {
if(head == null)
return null;
ListNode fast = head;
ListNode slow = head;
ListNode prev =null;
while(fast != null && fast.next != null)
{
fast = fast.next.next;
prev =slow;
slow=slow.next;
}
TreeNode root = new TreeNode(slow.val);
if(prev != null)
prev.next = null;
else
head = null;
root.left = sortedListToBST(head);
root.right = sortedListToBST(slow.next);
return root;
}
名字起的好高端快慢指针。但是算法复杂度巨高。一个点要遍历很多遍。。看来我的算法还是比较高效的。。