[LintCode/LeetCode] Convert Sorted List to Balanced BST [二叉搜索树]

最新推荐文章于 2023-04-13 15:01:12 发布

weixin_34109408

最新推荐文章于 2023-04-13 15:01:12 发布

阅读量101

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：https://segmentfault.com/a/1190000005022151

本文介绍了一种将有序单链表转换为高度平衡的二叉搜索树的方法。通过递归方式，先计算链表长度，再利用中点划分的方式构建平衡二叉树，确保树的高度尽可能均衡。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Problem

Given a singly linked list where elements are sorted in ascending order, convert it to a height balanced BST.

Example

               2
1->2->3  =>   / \
             1   3

Note

用递归的方法来做，首先新建cur结点来复制head结点，然后计算链表head的长度，调用helper(start, end)函数建立平衡BST。
当链表为空时，helper()中出现start大于end，返回null。然后计算中点mid，以mid为界分别递归构建左右子树。顺序是，左子树-->根结点-->右子树。由于根节点root是直接取cur.val构建，当前的cur已经被取用。所以在下一次递归构建右子树之前，要让cur指向cur.next。最后将root和左右子树相连，返回root。

Solution

public class Solution {
    ListNode cur;
    public TreeNode sortedListToBST(ListNode head) {  
        cur = head;
        int len = 0;
        while (head != null) {
            head = head.next;
            len++;
        }
        return helper(0, len-1);
    }
    public TreeNode helper(int start, int end) {
        if (start > end) return null;
        int mid = start+(end-start)/2;
        TreeNode left = helper(start, mid-1);
        TreeNode root = new TreeNode(cur.val);
        cur = cur.next;
        TreeNode right = helper(mid+1, end);
        root.left = left;
        root.right = right;
        return root;
    }
}