题目链接:https://leetcode-cn.com/problems/convert-sorted-list-to-binary-search-tree/
题目描述
给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9],
一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
思路
1 递归
当前方法和下一个方法的主要思路是:
给定列表中的中间元素将会作为二叉搜索树的根,该点左侧的所有元素递归的去构造左子树,同理右侧的元素构造右子树。这必然能够保证最后构造出的二叉搜索树是平衡的。
算法
- 由于我们得到的是一个有序链表而不是数组,我们不能直接使用下标来访问元素。我们需要知道链表中的中间元素。
- 我们可以利用两个指针来访问链表中的中间元素。假设我们有两个指针
slow
和fast
。slow
每次向后移动一个节点而fast
每次移动两个节点。当fast
到链表的末尾时slow
就访问到链表的中间元素。对于一个偶数长度的数组,中间两个元素都可用来作二叉搜索树的根。 - 当找到链表中的中间元素后,我们将链表从中间元素的左侧断开,做法是使用一个
prev
的指针记录slow
之前的元素,也就是满足prev->next = slow
。断开左侧部分就是让prev->next = nullptr
。 - 我们只需要将链表的头指针传递给转换函数,进行高度平衡二叉搜索树的转换。所以递归调用的时候,左半部分我们传递原始的头指针;右半部分传递
slow->next
作为头指针。
复杂度分析
- 时间复杂度: O ( N log N ) O(N \log N) O(NlogN)
- 空间复杂度: O ( log N ) O(\log N) O(logN)。因为使用递归的方法,所以需要考虑递归栈的空间复杂度。对于一棵费平衡二叉树,可能需要 O(N) 的空间,但是问题描述中要求维护一棵平衡二叉树,所以保证树的高度上界为 O(logN),因此空间复杂度为 O(logN)。
class Solution {
public:
TreeNode* sortedListToBST(ListNode* head) {
if(!head) return nullptr;