当前位置: 首页 > 文档资料 > LeetCode 题解 >

Tree - Convert Sorted List/Array to Binary Search Tree

优质
小牛编辑
134浏览
2023-12-01

Given a singly linked list where elements are sorted in ascending order, convert it to a height balanced BST.

这题需要将一个排好序的链表转成一个平衡二叉树,我们知道,对于一个二叉树来说,左子树一定小于根节点,而右子树大于根节点。所以我们需要找到链表的中间节点,这个就是根节点,链表的左半部分就是左子树,而右半部分则是右子树,我们继续递归处理相应的左右部分,就能够构造出对应的二叉树了。

这题的难点在于如何找到链表的中间节点,我们可以通过fast,slow指针来解决,fast每次走两步,slow每次走一步,fast走到结尾,那么slow就是中间节点了。

代码如下:

  1. class Solution {
  2. public:
  3. TreeNode *sortedListToBST(ListNode *head) {
  4. return build(head, NULL);
  5. }
  6. TreeNode* build(ListNode* start, ListNode* end) {
  7. if(start == end) {
  8. return NULL;
  9. }
  10. ListNode* fast = start;
  11. ListNode* slow = start;
  12. while(fast != end && fast->next != end) {
  13. slow = slow->next;
  14. fast = fast->next->next;
  15. }
  16. TreeNode* node = new TreeNode(slow->val);
  17. node->left = build(start, slow);
  18. node->right = build(slow->next, end);
  19. return node;
  20. }
  21. };

Convert Sorted Array to Binary Search Tree

Given an array where elements are sorted in ascending order, convert it to a height balanced BST.

这题类似上面那题,同样地解题方式,对于数组来说,能更方便的得到中间节点,代码如下:

  1. class Solution {
  2. public:
  3. TreeNode *sortedArrayToBST(vector<int> &num) {
  4. return build(num, 0, num.size());
  5. }
  6. TreeNode* build(vector<int>& num, int start, int end) {
  7. if(start >= end) {
  8. return NULL;
  9. }
  10. int mid = start + (end - start) / 2;
  11. TreeNode* node = new TreeNode(num[mid]);
  12. node->left = build(num, start, mid);
  13. node->right = build(num, mid + 1, end);
  14. return node;
  15. }
  16. };