您的位置:首页 > 其它

【leetcode】Convert Sorted List to Binary Search Tree

2015-06-02 22:12 381 查看

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);
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: