109. 有序链表转换二叉搜索树
给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
思路
由于链表是没有索引的,所以二分有些难以实现,这里我们使用快慢指针的方法可以有效的解决,给快指针调速至慢指针的二倍从而达到getMid的效果实现分治。
代码
/**
* 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:
ListNode* getMid(ListNode* left, ListNode* right) {
ListNode* fast, * slow;
fast = slow = left;
// fast两倍速,slow一倍速
while (fast != right && fast->next != right) {
fast = fast->next->next;
slow = slow->next;
}
return slow;
}
TreeNode* build(ListNode* left, ListNode* right) {
if (left == right) return nullptr;
auto mid = getMid(left, right);
auto root = new TreeNode(mid->val, build(left, mid), build(mid->next, right));
return root;
}
TreeNode* sortedListToBST(ListNode* head) {
return build(head, nullptr);
}
};