Java實現 LeetCode 109 有序鏈表轉換二叉搜索樹

109. 有序鏈表轉換二叉搜索樹

給定一個單鏈表，其中的元素按升序排序，將其轉換爲高度平衡的二叉搜索樹。

本題中，一個高度平衡二叉樹是指一個二叉樹每個節點 的左右兩個子樹的高度差的絕對值不超過 1。

示例:

給定的有序鏈表： [-10, -3, 0, 5, 9],

一個可能的答案是：[0, -3, 9, -10, null, 5], 它可以表示下面這個高度平衡二叉搜索樹：

      0
     / \
   -3   9
   /   /
 -10  5

 
class Solution {
     public TreeNode sortedListToBST(ListNode head) {
        if(head == null) return null;
        else if(head.next == null) return new TreeNode(head.val);
        ListNode pre = head;
        ListNode p = pre.next;
        ListNode q = p.next;
        //找到鏈表的中點p
        while(q!=null && q.next!=null){
            pre = pre.next;
            //p走一倍的速度
            p = pre.next;
            //q走兩倍的速度
            q = q.next.next;
        }
        //將中點左邊的鏈表分開
        pre.next = null;
        TreeNode root = new TreeNode(p.val);
        root.left = sortedListToBST(head);
        root.right = sortedListToBST(p.next);
        return root;
    }
}