题目:
Given a singly linked list where elements are sorted in ascending order, convert it to a height balanced BST.
题解:将一个有序链表转成二叉排序树,如果是一棵相对平衡的排序树,应该是这样的,链表中间那个节点为树的根节点,根节点的左子树节点应该是根节点左边那部分的中间节点,根节点的右节点应该是根节点右边那部分链表的中间节点,后面就按照这个规律依次类推了。知道上面的规律之后,问题的关键点就是寻找一个链表的中间节点了,这个应该是很熟悉的,用快慢两个指针可以轻松解决。下面是这种思路的实现代码:
代码:
//类似归并排序 O(Nlogn) public static TreeNode sortedListToBST(ListNode head) { ListNode fast=head; ListNode slow=head; if(head==null)//递归结束条件 return null; ListNode leftOfMid=null;//用来记录中间节点的前一个节点,这里把slow指向的节点看做中间及诶单 while(fast.next!=null&&fast.next.next!=null) { fast=fast.next.next; leftOfMid=slow; slow=slow.next; } if(leftOfMid==null)//说明没有进入上面的while循环 说明这时候只有1个或两个节点 这时候就不存在左链表的左半部分了 head=null; else leftOfMid.next=null;//如果有左半部分则把左半部分链表的末尾指向空,将链表截两段 ListNode rightOfMid=slow.next;//链表右半部分的开始 TreeNode root=new TreeNode(slow.val); root.left=sortedListToBST(head); root.right=sortedListToBST(rightOfMid); return root; }
上面的时间复杂区是n*logn,有没有O(n)的算法呢,新建每一个树节点一遍下来时间复杂度就是o(n)了,所以在寻找中间节点的时候只能是常数时间,这样才能让最终的时间复杂度为O(n),这样只能借助哈希表进行辅助了
代码:
//用哈希表O(n)时间复杂度 public static TreeNode sortedListToBST2(ListNode head) { Hashtable< Integer, TreeNode> hashtable=new Hashtable<>(); if(head==null) return null; ListNode cur=head; int length=0; while(cur!=null) { TreeNode node=new TreeNode(cur.val); hashtable.put(length,node ); cur=cur.next; length++; } return buildTree(0, length-1, hashtable); } public static TreeNode buildTree(int start,int end,Hashtable<Integer, TreeNode> hashtable) { if(start<=end) { int mid=(start+end)/2; TreeNode root=hashtable.get(mid); root.left=buildTree(start,mid-1 , hashtable); root.right=buildTree(mid+1, end, hashtable); return root; } else { return null; } }
版权声明:本文为博主原创文章,未经博主允许不得转载。
时间: 2024-10-10 10:52:41