LeetCode108 将有序数组转换为二叉搜索树

最新推荐文章于 2024-10-19 15:29:34 发布

红毛乌龟

最新推荐文章于 2024-10-19 15:29:34 发布

阅读量204

点赞数 3

分类专栏：算法刷题文章标签： c++ leetcode 数据结构算法

本文链接：https://blog.csdn.net/daishabby2486/article/details/141254098

版权

算法刷题专栏收录该内容

90 篇文章 0 订阅

订阅专栏

前言

题目： 108. 将有序数组转换为二叉搜索树
文档： 代码随想录——将有序数组转换为二叉搜索树
编程语言： C++
解题状态： 成功解答！

思路

从中位数还是递归，这样可以避免二叉树不平衡。

代码

/**
 * 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:
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        if (nums.size() == 0) return nullptr;
        int size = nums.size();
        TreeNode* root = new TreeNode(nums[size / 2]);
        if (size / 2 > 0) {
            vector<int> leftNums(nums.begin(), nums.begin() + nums.size() / 2);
            root -> left = sortedArrayToBST(leftNums);
        }

        if (size / 2 < size - 1) {
            vector<int> rightNums(nums.begin() + nums.size() / 2 + 1, nums.end());
            root -> right = sortedArrayToBST(rightNums);
        }
        
        return root;
    }
};