给定一个单链表/数组,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9],
一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
通过次数44,798提交次数59,804
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-sorted-list-to-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
1.数组方式,也是108题,对于一个有序数组,构建完全平衡二叉树关键在于递归的找到一段数组的中点,作为当前的根节点,之后向下递归,当前的根节点的左子树和右子树分别在数组中点的左边和右边,递归的求解即可;
2.对于有序链表,无法在O(1)时间复杂度找到中点,因此考虑进行优化,可以从数组方式中寻找突破口,可以知道,最后构建出来的树的中序遍历即是有序的链表,那么能否在构建的过程中,按照中序遍历的顺序进行重建呢?答案是可以的,相较于数组方式,我们不一定需要在new出树节点的时候马上给他赋值,因为我们只能获取当前构建的子树的起点和终点,要获取值需要进行链表的遍历,但是我们可以通过起点和终点判断出当前是否是叶子节点,换句话说,我们可以虚拟出一个包含n个节点的完全平衡二叉树,对其进行中序遍历并赋值,只不过遍历的时候也要一遍创建节点。
3.需要注意,对于指针,也需要传递指针引用,否则指针按值传递的话,p = p->next不会被记录。
/**
* 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) {
int len = 0;
ListNode* p = head;
while(p)
{
len++;
p = p->next;
}
return inorder(head,0,len-1);
}
TreeNode* inorder(ListNode*& head,int left,int right)
{
if(left>right)
{
return nullptr;
}
int mid = (left+right)/2;
TreeNode* cur = new TreeNode();
cur->left = inorder(head,left,mid-1);
cur->val = head->val;
head = head->next;
cur->right = inorder(head,mid+1,right);
return cur;
}
};