Leetcode 108. Convert Sorted Array to Binary Search Tree 给定排序数组,转换成平衡二叉搜索树 C++实现 nullptr使用

  • 二叉搜索树不作过多解释
  • 平衡二叉树是任意父结点的左右子树高度差不超过1
  • 因此算法的巧妙之处在于,通过二分的方法,每次取中间的数作为父结点,两侧数分别放到左右子树中。当遍历到最后时,左右子树要么都有,要么只有一个,要么都无,这几种情况均是平衡的
/**
 * 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) {
        return build(nums, 0, nums.size()-1);
    }
    
private:
    TreeNode* build(const vector<int>& nums, int l, int r){
        if(l>r) return nullptr;
        int m = l+(r-l)/2;
        TreeNode *root = new TreeNode(nums[m]);
        root->left = build(nums, l,m-1);
        root->right = build(nums, m+1, r);
        return root;
    }
};

猜你喜欢

转载自blog.csdn.net/Xurui_Luo/article/details/106729783