有序链表转换二叉搜索树
题目:
给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9],
一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
解题思路:因为输入数据是升序排列,所以找到中位数即可,找到中位数后确立根节点,则中位数以前的都是左节点,剩下的都是右节点,子节点也是如此
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public TreeNode sortedListToBST(ListNode head) {
if(head == null) {
return null;
}
return build(head, null);
}
private TreeNode build(ListNode start, ListNode end) {
if(start == end) {
return null;
}
ListNode mid = getMid(start, end);
TreeNode node = new TreeNode(mid.val);
node.left = build(start, mid);
node.right = build(mid.next, end);
return node;
}
private ListNode getMid(ListNode start, ListNode end) {
ListNode fast = start, slow = start;
while(fast.next != end && fast.next.next != end) {
fast = fast.next.next;
slow = slow.next;
}
return slow;
}
}