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

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

题目

给你一个整数数组 nums ，其中元素已经按 升序 排列，请你将其转换为一棵 高度平衡 二叉搜索树。高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。

思路

• 每次取中间位置的元素构造节点，TreeNode* root = new TreeNode(nums[mid]);
• 接着划分区间，root的左孩子接住下一层左区间的构造节点，右孩子接住下一层右区间构造的节点
• 最后返回root节点

代码

/**
 * 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* traversal(vector<int>& nums,int left,int right)
    {
        if(left > right)
        {
            return nullptr;
        }
        int mid = left + ((right - left) / 2);
        TreeNode* root = new TreeNode(nums[mid]);//左右区间的分割点
        root->left = traversal(nums,left,mid -1);
        root->right = traversal(nums,mid + 1,right);
        return root;
    }

    TreeNode* sortedArrayToBST(vector<int>& nums) {
        TreeNode* root = traversal(nums,0,nums.size() - 1);
        return root;
    }
};