有序链表转换二叉搜索树

题目链接

题目描述:

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

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

示例:

给定的有序链表: [-10, -3, 0, 5, 9],

一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:

   0
  /  \
-3   9
 /     /
-10  5

思路:

设当前链表的左端点为left,右端点right,包含关系为「左闭右开」,即left 包含在链表中而right 不包含在链表中。我们希望快速地找出链表的中位数节点mid。

为什么要设定「左闭右开」的关系?由于题目中给定的链表为单向链表,访问后继元素十分容易,但无法直接访问前驱元素。因此在找出链表的中位数节点 \textit{mid}mid 之后,如果设定「左闭右开」的关系,我们就可以直接用(left,mid) 以及(mid.next,right) 来表示左右子树对应的列表了。并且,初始的列表也可以用(head,null) 方便地进行表示,其中null 表示空节点。

找出链表中位数节点的方法多种多样,其中较为简单的一种是「快慢指针法」。初始时,快指针fast 和慢指针slow 均指向链表的左端点left。我们将快指针fast 向右移动两次的同时,将慢指针slow 向右移动一次,直到快指针到达边界(即快指针到达右端点或快指针的下一个节点是右端点)。此时,慢指针对应的元素就是中位数。

在找出了中位数节点之后,我们将其作为当前根节点的元素,并递归地构造其左侧部分的链表对应的左子树,以及右侧部分的链表对应的右子树。

 

class Solution {
public:
    ListNode* getMedian(ListNode* left, ListNode* right) {
        ListNode* fast = left;
        ListNode* slow = left;
        while (fast != right && fast->next != right) {
            fast = fast->next;
            fast = fast->next;
            slow = slow->next;
        }
        return slow;
    }

    TreeNode* buildTree(ListNode* left, ListNode* right) {
        if (left == right) {
            return nullptr;
        }
        ListNode* mid = getMedian(left, right);
        TreeNode* root = new TreeNode(mid->val);
        root->left = buildTree(left, mid);
        root->right = buildTree(mid->next, right);
        return root;
    }

    TreeNode* sortedListToBST(ListNode* head) {
        return buildTree(head, nullptr);
    }
};

 

posted @ 2021-04-20 13:57  阿涅—Rachel  阅读(122)  评论(0)    收藏  举报