leetcode 109. 有序链表转换二叉搜索树

109. 有序链表转换二叉搜索树

给定一个单链表，其中的元素按升序排序，将其转换为高度平衡的二叉搜索树。本题中，一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例: 给定的有序链表： [-10, -3, 0, 5, 9],
一个可能的答案是：[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树：

分析

主要考察的点就是链表的二分。

代码

class Solution {
public:
    //寻找链表的中间结点
    ListNode* getMid(ListNode* head) {
        //申请快慢指针
        ListNode* fast = head;
        ListNode* slow = head;
        ListNode* pre = head;
        while(fast && fast->next){
            pre = slow;
            slow = slow->next;
            fast = fast->next->next;
        }
        return pre;       
    }
    TreeNode* sortedListToBST(ListNode* head) {
        if(!head) return NULL;
        ListNode* pre_node = getMid(head);
        //pre_node的next结点作为根节点
        TreeNode* root;
        if(pre_node->next){
            root = new TreeNode(pre_node->next->val);
            ListNode* nxt = pre_node->next->next;
            pre_node->next = NULL;
            root->left = sortedListToBST(head);
            root->right = sortedListToBST(nxt);
        }else{
            root = new TreeNode(head->val);
        }     
        return root;        
    }
};

类似的题目

leetcode 108 将一个有序的数组转化为平衡的二叉搜索树。

分析

同样是使用二分，只是相对于链表的求解来说，直接求数组的中间索引即可。

代码

class Solution {
public:
    TreeNode* help(vector<int>& nums, int l, int r){
        //使用二分的方式，每次都去找子数组的中间元素
        if(l>r) return NULL;
        int mid = (l + r+1)/2;//注意要选中间的偏右的结点
        TreeNode* root = new TreeNode(nums[mid]);
        root->left = help(nums, l, mid-1);
        root->right = help(nums, mid+1, r);
        return root;
    }
    TreeNode* sortedArrayToBST(vector<int>& nums) {       
        return help(nums, 0, nums.size()-1);        
    }
};