LeetCode 108.Convert Sorted Array to Binary Search Tree (将有序数组转换为二叉搜索树)

╰半夏微凉° 2022-05-09 04:56 141阅读 0赞

题目描述：

将一个按照升序排列的有序数组，转换为一棵高度平衡二叉搜索树。

本题中，一个高度平衡二叉树是指一个二叉树*每个节点* 的左右两个子树的高度差的绝对值不超过 1。

**示例:**

给定有序数组: [-10,-3,0,5,9],
    
    一个可能的答案是：[0,-3,9,-10,null,5]，它可以表示下面这个高度平衡二叉搜索树：
    
          0
         / \
       -3   9
       /   /
     -10  5

**AC C++ Solution:**

**递归法**：

把数组从中间分开，中间元素作为根节点，左边的元素作为左子树，右边的元素作为右子树。

/**
     * 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) {
            if(nums.size() == 0)    return NULL;
            if(nums.size() == 1)
                return new TreeNode(nums[0]);
            
            int middle = nums.size()/2;
            TreeNode* root = new TreeNode(nums[middle]);    //中间元素作为根节点   
            vector<int> leftInts(nums.begin(),nums.begin()+middle);    //左边元素作为左子树   
            vector<int> rightInts(nums.begin()+middle+1,nums.end());    //右边元素作为右子树
            root->left = sortedArrayToBST(leftInts);
            root->right = sortedArrayToBST(rightInts);
            
            return root;
        }
    };