题目链接:leetcode109
题面
题目大意
给你一个有序数组,让你转成一个平衡的二叉搜索树
解题思路
乍一看这道题似乎要构造一个 平衡二叉树,而平衡二叉树的实现方法有多种,例如: SplayTree、AVLTree、SBTree、RBTree等。但代码量感人,根据他人思路大致总结如下几种方法:
寻找中位数+分治
具体思路就是每次对于一个有序链表,取中位数作为根节点,之后左半边和右半边分别递归构造子树。能否满足平衡的限制呢?我们发现左右区间的元素数量最多相差 1 个,显然满足。如何寻找中位数呢?数组中我们可以直接用下标索引,而链表中可以用 双指针 法进行操作,具体操作是一个指针走两步、另一个走一步。
时间复杂度 O ( n l o g n ) O(nlogn) O(nlogn),空间复杂度 O ( l o g n ) O(logn) O(logn)。
分治优化
同法一类似,具体优化的部分是每次我们没必要先去找那个中位数,而是先处理左子树的构造,等到左子树构造完毕后处理根节点自然链表 cur 走到了中位数的位置,之后转而处理右子树。也就是说链表的 cur 指针需要是一个引用传递。
时间复杂度 O ( n ) O(n) O(n),空间复杂度 O ( l o g n ) O(logn) O(logn)。
bfs+dfs建树
实际上,对于树的结构我们只要保证平衡即可,但是具体的构造我们并不太关心。如果我们能够先把树建起来,然后因为已经知道中序遍历的序列,我们能比较容易地把值填进去。
于是,我们考虑将树建成一个完全二叉树,显然,完全二叉树是平衡地,bfs可以帮我们到这件事。至于把数字填到树中可以借助 dfs 中序遍历二叉树。
时间复杂度 O ( n ) O(n) O(n),空间复杂度 O(n)。
代码实现
分治优化
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* sortedListToBST(ListNode* head) {
ListNode* p = head;
return build(0, getLength(p), p);
}
int getLength(ListNode* head) {
int len = 0;
while (head != NULL) {
len++;
head = head->next;
}
return len;
}
//[L, R)
TreeNode* build(int L, int R, ListNode*& _head) {
if (L >= R) return NULL;
TreeNode* _root = new TreeNode();
int mid = L+(R-L>>1);
_root->left = build(L, mid, _head);
_root->val = _head->val;
_head = _head->next;
_root->right = build(mid + 1, R, _head);
return _root;
}
};