Convert Sorted List to Binary Search Tree

最新推荐文章于 2020-08-18 08:24:28 发布

原创最新推荐文章于 2020-08-18 08:24:28 发布 · 427 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#搜索 #二叉树 #链表 #递归

Qt学习笔记专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种将已排序链表高效转换为平衡二叉搜索树的方法。通过寻找链表中点作为根节点，并递归地构建左右子树，确保了树的高度尽可能保持对数级别。

一、概述

平衡二叉树是一颗n个节点的高度为lgn的二叉查找树，严格维持节点的左右节点数差的绝对值小于等于1

对于一个已经排好序的链表来说，只需要找到中间点，把其当做root，其左边链表为root的left节点，其右边链表为root的right节点，递归构造就行

二、代码

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
public:
    TreeNode* sortedListToBST(ListNode* head) {
        TreeNode *root = NULL;
        if (head != NULL) {
            ListNode *slow, *fast, *pre;
            slow = fast = head;
            pre = NULL;
            while (fast->next && fast->next->next) {
                pre = slow;
                slow = slow->next;
                fast = fast->next->next;
            }
            if (pre != NULL) pre->next = NULL;
            else head = NULL;
            root = new TreeNode(slow->val);
            root->left = sortedListToBST(head);
            root->right = sortedListToBST(slow->next);
        }
        return root;
    }
};