[LeetCode] 109. 有序链表转换二叉搜索树 ☆☆☆(递归)

Posted 2021-03-18 反光的小鱼儿

描述

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

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

示例:

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

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

0
/
-3 9
/ /
-10 5

解析

有序链表，平衡二叉树，很自然想到将链表一分为二，两部分就是中间节点的左右节点。递归执行左右2个部分的节点即可。

代码

递归

/**
 * 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 {
    //me
    public TreeNode sortedListToBST(ListNode head) {
        if (null == head) {
            return null;
        } else if (null == head.next) {
            return new TreeNode(head.val);
        }
        ListNode slow = head;
        ListNode fast = head;
        ListNode pre = head;
        while (fast != null && fast.next != null) {
            pre = slow;
            slow = slow.next;
            fast = fast.next.next;
        }
        pre.next = null;
        TreeNode root = new TreeNode(slow.val);
        root.left = sortedListToBST(head);
        root.right = sortedListToBST(slow.next);
        return root;
    }
}

优化点

可以将有序链表转为有序数组，降低查询复杂度。空间换时间解法。

有序链表转为搜索二叉树