面试题 04.02. 最小高度树

最新推荐文章于 2023-03-12 16:38:26 发布

子言慕雨

最新推荐文章于 2023-03-12 16:38:26 发布

阅读量97

点赞数

分类专栏：程序员面试金典（第 6 版）文章标签： leetcode

本文链接：https://blog.csdn.net/TY_GYY/article/details/124425537

版权

程序员面试金典（第 6 版）专栏收录该内容

38 篇文章 0 订阅

订阅专栏

题

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/minimum-height-tree-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

代码

&符号是指取地址符，是单目运算符。当你定义一个变量a，那想知道变量在内存中的地址，就可以使用这个符号。
即使是指针 int *p 中的p只是值如果想取p的地址要用&p 一般用于作为函数的形式参数例如int * &p

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);
    }
};