1.题目
给定一个单元链表,元素按升序排序,将其转换为高度平衡的BST。
2.分析思路
思路来自牛客网JacobGo!
这道题是要求把有序链表转为二叉搜索树,由于二分查找法每次需要找到中点,而链表的查找中间点可以通过快慢指针来操作。找到中点后,要以中点的值建立一个数的根节点,然后需要把原链表断开,分为前后两个链表,都不能包含原中节点,然后再分别对这两个链表递归调用原函数,分别连上左右子节点即可。
3.代码实现
(1)代码1
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; next = null; }
* }
*/
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public TreeNode sortedListToBST(ListNode head) {
return toBST(head, null);
}
private TreeNode toBST(ListNode head, ListNode tail) {
if (head == tail)
return null;
// 申请两个指针,fast移动速度是low的两倍
ListNode fast = head;
ListNode slow = head;
while (fast != tail && fast.next != tail) {
fast = fast.next.next;
slow = slow.next;
}
TreeNode root = new TreeNode(slow.val);
root.left = toBST(head, slow);
root.right = toBST(slow.next, tail);
return root;
}
(2)代码2:
/**
*用一个表List存储链表中所有的n个ListNode元素。
*然后以中间元素(第n/2个元素)的值创建根节点,
*将List分为两部分,以前半部分(1...n/2-1)的中间节点的值创建节点,为根节点的左孩子,
*以后半部分(n/2+1...n)的中间节点的值创建节点,为根的右孩子。
*递归重复实现上述步骤
*/
import java.util.*;
public class Solution {
public TreeNode sortedListToBST(ListNode head) {
if(head==null) return null;
ArrayList<ListNode> list=new ArrayList<>();
while(head!=null){
list.add(head);
head=head.next;
}
TreeNode root=sortedListToBST(list,0,list.size()-1);
return root;
}
public TreeNode sortedListToBST(ArrayList<ListNode> list,int start,int end){
if(start>end) return null;
TreeNode node=new TreeNode(list.get((start+end+1)/2).val);
node.left=sortedListToBST(list,start,(start+end+1)/2-1);
node.right=sortedListToBST(list,(start+end+1)/2+1,end);
return node;
}
}
题目来自牛客网leetcode