LintCode-排序列表转换为二分查找树

给出一个所有元素以升序排序的单链表,将它转换成一棵高度平衡的二分查找树

您在真实的面试中是否遇到过这个题?  Yes
样例
标签 
Expand  

相关题目 
Expand 

分析:就是一个简单的递归,只是需要有些链表的操作而已

代码:

/**
 * Definition of ListNode
 * class ListNode {
 * public:
 *     int val;
 *     ListNode *next;
 *     ListNode(int val) {
 *         this->val = val;
 *         this->next = NULL;
 *     }
 * }
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param head: The first node of linked list.
     * @return: a tree node
     */
    TreeNode *sortedListToBST(ListNode *head) {
        // write your code here
        if(head==nullptr)
            return nullptr;
        int len = 0;
        ListNode*temp = head;
        while(temp){len++;temp = temp->next;};
        if(len==1)
        {
            return new TreeNode(head->val);
        }
        else if(len==2)
        {
            TreeNode*root = new TreeNode(head->val);
            root->right = new TreeNode(head->next->val);
            return root;
        }
        else
        {
            len/=2;
            temp = head;
            int cnt = 0;
            while(cnt<len)
            {
                temp = temp->next;
                cnt++;
            }
            ListNode*pre = head;
            while(pre->next!=temp)
                pre = pre->next;
            pre->next = nullptr;
            TreeNode*root = new TreeNode(temp->val);
            root->left = sortedListToBST(head);
            root->right = sortedListToBST(temp->next);
            return root;
            
        }
    }
};


    原文作者:LintCode题目解答
    原文地址: https://blog.csdn.net/wangyuquanliuli/article/details/47359405
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞