【C++】【LeetCode】108. Convert Sorted Array to Binary Search Tree

zjajgyy

于 2017-08-12 11:36:53 发布

阅读量816

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/zjajgyy/article/details/77113302

LeetCode 专栏收录该内容

51 篇文章

订阅专栏

本文介绍了一种将升序数组转换为高度平衡的二叉搜索树的方法。通过选取数组的中位数作为树的根节点，并递归地构建左右子树，确保了树的高度平衡。

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

题目

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

思路

目的是要得到一个平衡二叉树，也就说是每棵数的根节点应该是整棵树上结点的中位数。所以取数组的中位数作为根节点，中位数左边为左子树的数组，右边为右子数的数组。以此类推。

代码

/**
 * 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* sortedArrayToBST(vector<int>& nums) {
        TreeNode* root;
        root = helper(root, nums, 0, nums.size() - 1);
        return root;
    }

    TreeNode* helper(TreeNode* root, vector<int>& nums, int start, int end) {
        if (start > end) {
            return NULL;
        }
        root = new TreeNode(0);
        root->val = nums[(start + end + 1) / 2];
        root->left = helper(root->left, nums, start, (start + end + 1) / 2 - 1);
        root->right = helper(root->right, nums, (start + end + 1) / 2 + 1, end);
        return root;
    }
};