0
点赞
收藏
分享

微信扫一扫

面试题 04.02. 最小高度树

犹大之窗 2022-04-26 阅读 75
leetcode

给定一个有序整数数组,元素各不相同且按升序排列,编写一个算法,创建一棵高度最小的二叉搜索树。

示例:
给定有序数组: [-10,-3,0,5,9],

一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:

      0 
     / \ 
   -3   9 
   /   / 
 -10  5 

代码

class Solution
{

public:
    TreeNode* sortedArrayToBST(vector<int> &nums)
    {
        if (nums.size()==0)
        {
            return NULL;
        }
        
        TreeNode *ans;
        sort(nums.begin(), nums.end());
        doIt(ans, nums);

        return ans;
    }

    void doIt(TreeNode* &node, vector<int> nums) 
    {
        if (nums.size() == 0)
        {
            return;
        }
        int middle = (nums.size() - 1) / 2;
        node = new TreeNode(nums[middle]);
        vector<int>l,r;
        for (int i = 0; i < nums.size(); i++)
        {
            if (i == middle)
            {
                continue;
            }
            
            if (i<middle)
            {
                l.push_back(nums[i]);
            }
            else
            {
                r.push_back(nums[i]);
            }
        }

        doIt(node->left, l);
        doIt(node->right, r);
    }
};
举报

相关推荐

0 条评论