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

最新推荐文章于 2024-04-07 15:58:50 发布

baixiaofei567

最新推荐文章于 2024-04-07 15:58:50 发布

阅读量102

点赞数

分类专栏： leetcode # 树文章标签： leetcode

本文链接：https://blog.csdn.net/J_avaSmallWhite/article/details/113733525

版权

AVL树二分递归高度平衡有序数组二叉搜索树

关键词由CSDN通过智能技术生成

leetcode 同时被 2 个专栏收录

343 篇文章 0 订阅

订阅专栏

树

110 篇文章 0 订阅

订阅专栏

在这里插入图片描述
还以为手写avl，以前pat的时候手写avl轻轻松松，现在估计有点难
其实就是二分递归建树，找到mid，用mid左边的序列来建左子树，mid右边来建右子树，mid用来创建当前结点。l>r才返回，=不返回。

/**
 * 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* helper(vector<int>& nums,int l, int r){
        if(l > r) return nullptr;
        int mid = l + (r-l)/2;
        TreeNode* root = new TreeNode(nums[mid]);
        root->left = helper(nums,l,mid-1);
        root->right = helper(nums,mid+1,r);
        return root;
    }  
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        //要高度平衡的BST，那不就是avl树？
        //用一个二分来选定每次的根结点
        if(nums.size() == 0) return nullptr;
        return helper(nums,0,nums.size()-1);
    }
};

因为要高度平衡，而且直接是有序的，所以我们用二分递归建树，递归边界就是l>r

class Solution {
public:
    TreeNode* helper(vector<int>& nums, int l, int r){
        if(l > r) return nullptr;
        int mid = l + (r-l)/2;
        TreeNode* root = new TreeNode(nums[mid]);
        root->left = helper(nums,l,mid-1);
        root->right = helper(nums,mid+1,r);
        return root;
    }
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        //bst的特点：中序是从小到大的，所以我们相当于已经有它的中序了
        //又因为要高度平衡，所以我们用二分，每次取中间作为根结点，然后递归搭建左右子树
        TreeNode* root = helper(nums,0,nums.size()-1);
        return root;
    }
};