[Leetcode] Convert Sorted List to BST

最新推荐文章于 2023-05-18 15:11:28 发布

原创最新推荐文章于 2023-05-18 15:11:28 发布 · 383 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

67 篇文章

订阅专栏

介绍如何将一个升序排列的单链表转换为高度平衡的二叉搜索树。采用递归方法，通过中序遍历的形式，选择中间节点作为根节点，确保左右子树的高度尽可能相同。

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

起初觉得这道题和 Convert Sorted Array to BST 是一样的，但是如果要用那个二分法的算法，则需要将 list 先处理成 array。

这样一来，原来的 O(logn) 又变成 O(n)，而且还需要维护额外的空间。

做法应该是这样的: http://leetcode.com/2010/11/convert-sorted-list-to-balanced-binary.html

这种做法是 bottom-to-top，原先的做法是 top-to-bottom。虽然复杂度仍然是 O(n)，但是却 neat 了不少。

代码有点中序遍历的 form。

BinaryTree* sortedListToBST(ListNode *& list, int start, int end) {
  if (start > end) return NULL;
  // same as (start+end)/2, avoids overflow
  int mid = start + (end - start) / 2;
  BinaryTree *leftChild = sortedListToBST(list, start, mid-1);
  BinaryTree *parent = new BinaryTree(list->data);
  parent->left = leftChild;
  list = list->next;
  parent->right = sortedListToBST(list, mid+1, end);
  return parent;
}
 
BinaryTree* sortedListToBST(ListNode *head, int n) {
  return sortedListToBST(head, 0, n-1);
}