LeetCode 109.Convert Sorted List to Binary Search Tree (有序链表转换二叉搜索树)

Bertha 。 2022-05-09 04:56 181阅读 0赞

题目描述：

给定一个单链表，其中的元素按升序排序，将其转换为高度平衡的二叉搜索树。

本题中，一个高度平衡二叉树是指一个二叉树*每个节点* 的左右两个子树的高度差的绝对值不超过 1。

**示例:**

给定的有序链表： [-10, -3, 0, 5, 9],
    
    一个可能的答案是：[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树：
    
          0
         / \
       -3   9
       /   /
     -10  5

AC C++ Solution:

本题解法和将有序数组转换为二叉搜索树的思路其实是一样，区别主要是查找中间元素。

递归法：

/**
     * 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* sortedListToBST(ListNode* head) {
            return sortedListToBST(head, NULL);
        }
        
    private:
        TreeNode *sortedListToBST(ListNode *head, ListNode *tail)
        {
            if(head == tail)
                return NULL;
            if(head->next == tail) {
                TreeNode *root = new TreeNode(head->val);
                return root;
            }
            
            ListNode *mid = head, *temp = head;
            while(temp != tail && temp->next != tail)   //寻找中间节点
            {
                mid = mid->next;    
                temp = temp->next->next;    //mid每前进一次，temp前进两次，当temp到达链表尾时，mid正好是中间元素
            }
            TreeNode *root = new TreeNode(mid->val);
            root->left = sortedListToBST(head,mid);
            root->right = sortedListToBST(mid->next,tail);
            return root;
        }
    };