LeetCode 109 有序链表转换二叉搜索树 Java

题目描述

给定一个单链表，其中的元素按升序排序，将其转换为高度平衡的二叉搜索树。

本题中，一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

示例:

给定的有序链表： [-10, -3, 0, 5, 9],

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

  0
 / \

-3 9
/ /
-10 5

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/convert-sorted-list-to-binary-search-tree
思路：和有序数组转换成二叉搜索树非常像，有序链表可以通过slow，fast两个指针找到中间节点，但也可以现将有序链表转化成数组。

代码

/**
 * 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; }
 * }
 */
class Solution {
    public TreeNode sortedListToBST(ListNode head) {
        if (head == null) return null;
        int len = 0;
        ListNode p = head;
        while(p != null){
            len++;
            p = p.next;
        }
        p = head;
        int[] nums = new int[len];
        int index = 0;
        while(p != null){
            nums[index] = p.val;
            index++;
            p = p.next;
        }
        return convertToTree(nums, 0, nums.length-1);
    }

    public TreeNode convertToTree(int[] nums, int left, int right){
        if (left <= right){
            int mid = (left + right) / 2;
            TreeNode newNode = new TreeNode(nums[mid]);
            newNode.left = convertToTree(nums, left, mid-1);
            newNode.right = convertToTree(nums, mid+1, right);
            return newNode;
        }else{
            return null;
        }
    }
}