Java实现LeetCode109.有序链表转换二叉搜索树【待优化版】

题目描述

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

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

思路

对于一个递增链表，要将其转换成高度平衡的二叉树，则需取其中点 mid 作为根节点，再对左右中点的左右链表[ left , mid )、[mid.next , right) 进行递归得到左右子树。

代码

class Solution {
    public TreeNode sortedListToBST(ListNode head) {
        TreeNode root = null;
        if( head==null ) return root;
        List<Integer> li = new ArrayList();
        while( head!=null ){
            li.add( head.val );
            head = head.next;
        }
        return dfs( li , 0 , li.size()-1 );
    }
    public TreeNode dfs ( List<Integer> li , int left , int right ){
        if( left > right) return null;
        if( left==right ) return new TreeNode( li.get(left) );
        int mid = ( left + right )/2;
        TreeNode root = new TreeNode( li.get( mid ) );
        root.left = dfs( li , left , mid-1 );
        root.right = dfs( li , mid + 1 , right );
        return root; 
    }
}

思考

在开始写的时候也想到了快慢指针取链表中点，但觉得改变区间后不好判断，就放弃了。

本代码的不足之处在于先将链表转化为了List，在List中不断地取中点、改变区间，从而导致较大的时间空间开销。