https://leetcode-cn.com/problems/convert-sorted-list-to-binary-search-tree/
思路一:高度平衡的二叉搜索树,即任意一个根节点均满足其左子树的高度与右子树的高度之差的绝对值 < = 1 <=1 <=1的二叉搜索树。那么每次都选择当前区间的中间节点作为根节点就可以做到平衡。如何选择中间节点呢?有两种办法: ( 1 ) (1) (1)直接从链表身上入手,利用快慢指针求中间节点。 ( 2 ) (2) (2)先把链表的各个节点存到数组中,然后可以 O ( 1 ) O(1) O(1)得到中间节点。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* dfs(vector<ListNode*>& vec,int l,int r){
if(r<=l)
return nullptr;
int mid=l+r>>1;
TreeNode *root=new TreeNode(vec[mid]->val);
root->left=dfs(vec,l,mid);
root->right=dfs(vec,mid+1,r);
return root;
}
TreeNode* sortedListToBST(ListNode* head) {
if(!head)
return nullptr;
vector<ListNode*> vec;
ListNode *tmp=head;
while(tmp){
vec.push_back(tmp);
tmp=tmp->next;
}
return dfs(vec,0,vec.size());
}
};
思路二:对一棵二叉搜索树做中序遍历(左根右)即可得到一个有序的数组(连续性)。于是我们可以模拟这个过程,动态的建树,这样就可以省略思路一中数组的空间开销。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
private:
ListNode *cur;
public:
TreeNode* dfs(int l,int r){
if(r<=l)
return nullptr;
int mid=l+r>>1;
TreeNode *left=dfs(l,mid);
TreeNode *root=new TreeNode(cur->val);
cur=cur->next;
root->left=left;
root->right=dfs(mid+1,r);
return root;
}
TreeNode* sortedListToBST(ListNode* head) {
if(!head)
return nullptr;
cur=head;
int n=0;
while(head&&++n)
head=head->next;
return dfs(0,n);
}
};