将有序数组转换为二叉搜索树(递归)

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

 

示例 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] 都是高度平衡二叉搜索树。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    // 辅助函数:将有序数组转换为二叉搜索树(BST)
    // 参数 nums: 有序整数数组
    // 参数 left: 当前子数组的左边界索引
    // 参数 right: 当前子数组的右边界索引
    TreeNode* sortedArrayToBST(vector<int>& nums, int left, int right) {
        // 如果左边界不大于右边界,则继续构建树节点
        if (left <= right) {
            // 计算中间位置
            int mid = left + (right - left) / 2;
            
            // 创建新的树节点,值为数组中间元素
            TreeNode* node = new TreeNode(nums[mid]);
            
            // 递归调用,构建当前节点的左子树,范围是[left, mid-1]
            node->left = sortedArrayToBST(nums, left, mid - 1);
            
            // 递归调用,构建当前节点的右子树,范围是[mid+1, right]
            node->right = sortedArrayToBST(nums, mid + 1, right);
            
            // 返回构建好的树节点
            return node;
        }
        
        // 如果左边界大于右边界,返回空指针,表示此路径没有更多节点
        return nullptr;
    }

    // 主函数:接收一个有序数组,并将其转换成高度平衡的二叉搜索树(BST)
    // 参数 nums: 有序整数数组
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        // 获取数组大小
        int n = nums.size();
        
        // 调用辅助函数,从整个数组开始构建BST,范围是[0, n-1]
        return sortedArrayToBST(nums, 0, n - 1);
    }
};

 

posted on 2024-12-24 17:35  _月生  阅读(63)  评论(0)    收藏  举报