题目描述:
给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9],
一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:
链接:https://leetcode-cn.com/problems/convert-sorted-list-to-binary-search-tree
思路分析:链表寻找中间值是很麻烦的,将链表中的值存储到vector<int>中,这样支持随机访问,然后进行构造即可。
代码:
class Solution {
public:
vector<int> v;//vector数组
TreeNode* FormBST(int start, int end)
{
if(start >= end)//终止条件
return nullptr;
int cur = (start + end) >> 1;//中间节点
TreeNode* root = new TreeNode(v[cur]);//根节点
root->left = FormBST(start,cur);//左子树
root->right = FormBST(cur + 1,end);//右子树
return root;
}
TreeNode* sortedListToBST(ListNode* head) {
if(head == nullptr)
return nullptr;
while(head != nullptr)
{
v.push_back(head->val);
head = head->next;
}
return FormBST(0,v.size());//构造bst,然后返回
}
};