Leetcode_convert-sorted-array-to-binary-search-tree

最新推荐文章于 2022-02-19 22:56:38 发布

原创最新推荐文章于 2022-02-19 22:56:38 发布 · 747 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

155 篇文章

订阅专栏

本文介绍如何将一个升序排列的数组转换为高度平衡的二叉搜索树。通过递归方式寻找中间节点作为根节点，并分别构建左右子树。提供两种实现方案，第一种采用整体分割的方式构建树结构，第二种则利用递归定义直接构造。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

地址：http://oj.leetcode.com/problems/convert-sorted-array-to-binary-search-tree/

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

思路：先找到头节点，再递归构造左子树和右子树。124ms。

参考代码：

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void construct_tree(TreeNode**p, vector<int>vec)
    {
        if(vec.empty())
            return;
        int mid = vec.size()/2;
        *p = new TreeNode(vec[mid]);
        vector<int>left_vec, right_vec;
        for(int i = 0; i<mid; ++i)
            left_vec.push_back(vec[i]);
        for(int i=mid+1; i<vec.size(); ++i)
            right_vec.push_back(vec[i]);
        construct_tree(&((*p)->left), left_vec);
        construct_tree(&((*p)->right), right_vec);
    }
    TreeNode *sortedArrayToBST(vector<int> &num) {
        if(num.empty())
            return NULL;
        TreeNode* head = NULL;
        construct_tree(&head, num);
        return head;
    }
};

SECOND TRIAL, 60MS

class Solution {
private:
    TreeNode* construct_tree(TreeNode* root, vector<int>& num, int begin, int end)
    {
        if(begin > end)
            return NULL;
        int mid = (begin+end)/2;
        root = new TreeNode(num[mid]);
        root->left = construct_tree(root->left, num, begin, mid-1);
        root->right = construct_tree(root->right, num, mid+1, end);
        return root;
    }
public:
    TreeNode *sortedArrayToBST(vector<int> &num) {
        return construct_tree(NULL, num, 0, num.size()-1);
    }
};