Convert Sorted List to Binary Search Tree
思路:
可以参考下Convert Sorted Array to Binary Search Tree。但是list无法直接访问下标,需要进行一些处理,可以通过自下到顶的方式来写。注意的是:
函数传递参数为ListNode* &head (仅存在在函数的传递参数中), 表示函数结束之后,head的修改之仍有效,而ListNode* head函数结束之后,head恢复原始情况,就没法使用了。
/**
* 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* convert(ListNode* &head, int s, int e)
{
if(s>e) return NULL;
int mid=s+(e-s)/2;
TreeNode* left=convert(head,s,mid-1);
TreeNode* root=new TreeNode(head->val);
root->left=left;
head=head->next;
root->right=convert(head,mid+1,e);
return root;
}
TreeNode* sortedListToBST(ListNode* head) {
if(head==NULL) return NULL;
int n=0;
ListNode *p=head;
while(p)
{
p=p->next;
n++;
}
return convert(head,0,n-1);
}
};