leetcode 108-将有序数组转换为二叉搜索树

xiyu_dang

于 2022-08-08 18:58:14 发布

阅读量56

点赞数

分类专栏：算法 # leetcode 文章标签：数据结构 c语言算法

本文链接：https://blog.csdn.net/qq_41555003/article/details/126233839

版权

算法同时被 2 个专栏收录

155 篇文章 4 订阅

订阅专栏

leetcode

87 篇文章 0 订阅

订阅专栏

给你一个整数数组 nums ，其中元素已经按升序排列，请你将其转换为一棵高度平衡二叉搜索树。

高度平衡二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。

示例 1：

输入：nums = [-10,-3,0,5,9]
输出：[0,-3,9,-10,null,5]
解释：[0,-10,5,null,-3,null,9] 也将被视为正确答案：

示例 2：

输入：nums = [1,3]
输出：[3,1]
解释：[1,null,3] 和 [3,1] 都是高度平衡二叉搜索树。

提示：

1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 按严格递增顺序排列

思路：

二分 + 递归

将数组内位置在中间的数据当作根，然后递归定义左子树和右子树；

因为题目只要求时高度平衡二叉树，所以当数组长度为偶数时取 numSize/2 位置上的元素即可

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

//二分 + 递归
struct TreeNode* sortedArrayToBST(int* nums, int numsSize){
    if(numsSize < 1)
        return NULL;
    struct TreeNode *root = (struct TreeNode*)malloc(sizeof(struct TreeNode));
    int mid = numsSize / 2;
    root->val = nums[mid];
    root->left = sortedArrayToBST(nums, mid);
    root->right =sortedArrayToBST(nums+mid+1, numsSize - mid - 1);
    return root;
}